diff --git a/vllm/model_executor/layers/rotary_embedding.py b/vllm/model_executor/layers/rotary_embedding.py index 7fa6c5e7fcde4..d4e9ed87ed54f 100644 --- a/vllm/model_executor/layers/rotary_embedding.py +++ b/vllm/model_executor/layers/rotary_embedding.py @@ -990,7 +990,7 @@ def get_rope( base, is_neox_style, dtype, short_factor, long_factor, **extra_kwargs) elif scaling_type == "mrope": - return MRotaryEmbedding( + rotary_emb = MRotaryEmbedding( head_size, rotary_dim, max_position,