From e20eba753bbced43837aa92f747e6c50ee36ce09 Mon Sep 17 00:00:00 2001 From: Mengqing Cao Date: Fri, 17 Oct 2025 17:00:30 +0800 Subject: [PATCH] [VLM][Refactor] Remove useless func `get_input_positions` in `MRotaryEmbedding` (#27088) Signed-off-by: MengqingCao --- .../layers/rotary_embedding/mrope.py | 34 ------------------- 1 file changed, 34 deletions(-) diff --git a/vllm/model_executor/layers/rotary_embedding/mrope.py b/vllm/model_executor/layers/rotary_embedding/mrope.py index 5cae3d9b80..d269733083 100644 --- a/vllm/model_executor/layers/rotary_embedding/mrope.py +++ b/vllm/model_executor/layers/rotary_embedding/mrope.py @@ -4,7 +4,6 @@ import numpy as np import torch -from transformers import PretrainedConfig from vllm.triton_utils import tl, triton @@ -376,39 +375,6 @@ class MRotaryEmbedding(RotaryEmbedding): ) -> tuple[torch.Tensor, torch.Tensor | None]: return self.forward_native(positions, query, key, offsets) - @classmethod - def get_input_positions( - cls, - input_tokens: list[int], - hf_config: PretrainedConfig, - image_grid_thw: list[list[int]] | torch.Tensor | None, - video_grid_thw: list[list[int]] | torch.Tensor | None, - second_per_grid_ts: list[float] | None, - context_len: int = 0, - seq_len: int | None = None, - audio_feature_lengths: torch.Tensor | None = None, - use_audio_in_video: bool = False, - ) -> tuple[list[list[int]], int]: - """Get mrope input positions and delta value.""" - - image_grid_thw = [] if image_grid_thw is None else image_grid_thw - video_grid_thw = [] if video_grid_thw is None else video_grid_thw - second_per_grid_ts = [] if second_per_grid_ts is None else second_per_grid_ts - - llm_positions, mrope_position_delta = cls.get_input_positions_tensor( - input_tokens=input_tokens, - hf_config=hf_config, - image_grid_thw=image_grid_thw, - video_grid_thw=video_grid_thw, - second_per_grid_ts=second_per_grid_ts, - context_len=context_len, - seq_len=seq_len, - audio_feature_lengths=audio_feature_lengths, - use_audio_in_video=use_audio_in_video, - ) - - return llm_positions.tolist(), mrope_position_delta - @staticmethod def get_next_input_positions( mrope_position_delta: int,