diff --git a/python/sglang/multimodal_gen/runtime/models/encoders/qwen2_5vl.py b/python/sglang/multimodal_gen/runtime/models/encoders/qwen2_5vl.py index a7b5f8825..b7cd9e5f6 100644 --- a/python/sglang/multimodal_gen/runtime/models/encoders/qwen2_5vl.py +++ b/python/sglang/multimodal_gen/runtime/models/encoders/qwen2_5vl.py @@ -1069,18 +1069,17 @@ class Qwen2_5_VLModel(nn.Module): self.rope_deltas = rope_deltas else: batch_size, seq_length, _ = inputs_embeds.shape - position_ids = torch.arange(seq_length, device=inputs_embeds.device) - position_ids = position_ids.view(1, 1, -1).expand(3, batch_size, -1) if cache_position is not None: delta = (cache_position[0] + self.rope_deltas).to( inputs_embeds.device ) else: - delta = torch.zeros( - (batch_size, seq_length), device=inputs_embeds.device - ) - delta = delta.repeat_interleave(batch_size // delta.shape[0], dim=1) - position_ids += delta.to(position_ids.device) + delta = torch.zeros(batch_size, 1, device=inputs_embeds.device) + position_ids = ( + (torch.arange(seq_length, device=inputs_embeds.device) + delta) + .unsqueeze(0) + .expand(3, -1, -1) + ) outputs = self.language_model( input_ids=None,