From 6211413677ae96ca2af82efff9ca7130ba46bd16 Mon Sep 17 00:00:00 2001 From: vraiti Date: Thu, 2 Apr 2026 12:28:53 -0400 Subject: [PATCH] Update MRoPE config fallback logic (#2278) Signed-off-by: vraiti Co-authored-by: Canlin Guo --- vllm_omni/model_executor/layers/rotary_embedding/mrope.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/vllm_omni/model_executor/layers/rotary_embedding/mrope.py b/vllm_omni/model_executor/layers/rotary_embedding/mrope.py index 463e55507..3d3a88d87 100644 --- a/vllm_omni/model_executor/layers/rotary_embedding/mrope.py +++ b/vllm_omni/model_executor/layers/rotary_embedding/mrope.py @@ -337,12 +337,11 @@ class OmniMRotaryEmbedding(MRotaryEmbedding): """ thinker_config = hf_config.thinker_config - try: + if hasattr(thinker_config, "audio_token_index"): audio_token_id = thinker_config.audio_token_index image_token_id = thinker_config.image_token_index video_token_id = thinker_config.video_token_index - except Exception: - logger.info("Multimodal token idx changed!") + else: audio_token_id = thinker_config.audio_token_id image_token_id = thinker_config.image_token_id video_token_id = thinker_config.video_token_id