Skip to content

Commit e48bdcc

Browse files
committed
fix lint
Signed-off-by: Shanshan Shen <[email protected]>
1 parent 2b0e3e7 commit e48bdcc

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

vllm_ascend/patch/worker/patch_qwen3_vl.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -175,10 +175,11 @@ def rot_pos_emb(self, grid_thw: list[list[int]]):
175175
# Use pre-computed cos_sin_cache from RotaryEmbedding
176176
cos, sin = self.rotary_pos_emb.get_cos_sin(max_grid_size)
177177

178-
cos_h = cos[pos_ids[:, 0]] # (num_tokens, rotary_dim // 2)
179-
cos_w = cos[pos_ids[:, 1]]
180-
sin_h = sin[pos_ids[:, 0]]
181-
sin_w = sin[pos_ids[:, 1]]
178+
# (num_tokens, rotary_dim // 2)
179+
cos_h = cos[pos_ids[:, 0]] # type: ignore
180+
cos_w = cos[pos_ids[:, 1]] # type: ignore
181+
sin_h = sin[pos_ids[:, 0]] # type: ignore
182+
sin_w = sin[pos_ids[:, 1]] # type: ignore
182183

183184
cos_combined = torch.cat([cos_h, cos_w], dim=-1)
184185
sin_combined = torch.cat([sin_h, sin_w], dim=-1)

0 commit comments

Comments
 (0)