bugfix for mrope

Signed-off-by: zouyida <zouyida@huawei.com>
vllm-project · Feb 26, 2025 · dce979a · dce979a
1 parent d535c1d
commit dce979a
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/vllm_ascend/model_runner.py b/vllm_ascend/model_runner.py
@@ -721,14 +721,10 @@ def _compute_multi_modal_input(self, inter_data: InterDataForSeqGroup,
                 mrope_input_positions, mrope_position_delta = \
                     MRotaryEmbedding.get_input_positions(
                         token_ids,
+                        hf_config
                         image_grid_thw=image_grid_thw,
                         video_grid_thw=video_grid_thw,
-                        image_token_id=hf_config.image_token_id,
-                        video_token_id=hf_config.video_token_id,
-                        vision_start_token_id=hf_config.vision_start_token_id,
-                        vision_end_token_id=hf_config.vision_end_token_id,
-                        spatial_merge_size=hf_config.vision_config.
-                        spatial_merge_size,
+                        second_per_grid_ts=None,
                         context_len=inter_data.context_lens[seq_idx],
                         seq_len=inter_data.seq_lens[seq_idx],
                     )