fix lint introduced in #20708 (#20886)

2026-03-18 15:38:52 -07:00
parent 0d23a461a0
commit eea9e19c13
1 changed files with 1 additions and 4 deletions
--- a/python/sglang/srt/models/deepseek_v2.py
+++ b/python/sglang/srt/models/deepseek_v2.py
@@ -1217,10 +1217,7 @@ class DeepseekV2AttentionMLA(
            )

            if rope_scaling and rope_scaling.get("apply_yarn_scaling", True):
-                mscale_all_dim = rope_scaling.get("mscale_all_dim", False)
-                scaling_factor = rope_scaling["factor"]
-                mscale = yarn_get_mscale(scaling_factor, float(mscale_all_dim))
-                self.scaling = self.scaling * mscale * mscale
+                self.scaling = compute_mla_mscale_scaling(rope_scaling, self.scaling)
        else:
            self.rotary_emb = None
        self.use_deepseek_yarn_rope = rope_scaling is not None