diff --git a/vllm/model_executor/layers/rotary_embedding.py b/vllm/model_executor/layers/rotary_embedding.py index c9d21ed6e3f60..1285627ec3cc5 100644 --- a/vllm/model_executor/layers/rotary_embedding.py +++ b/vllm/model_executor/layers/rotary_embedding.py @@ -804,7 +804,7 @@ def get_rope( } rotary_emb = DeepseekScalingRotaryEmbedding( head_size, rotary_dim, original_max_position, base, - is_neox_style, scaling_factor, dtype, *extra_kwargs) + is_neox_style, scaling_factor, dtype, **extra_kwargs) # The correct one should be "longrope" but keep "su" here # for backward compatible elif scaling_type == "su" or scaling_type == "longrope":