minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

fix lint introduced in #20708 (#20886)

Browse Source
This commit is contained in:
Qiaolin Yu
2026-03-18 15:38:52 -07:00
committed by GitHub
parent 0d23a461a0
commit eea9e19c13
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/models/deepseek_v2.py
+1 -4
View File
@@ -1217,10 +1217,7 @@ class DeepseekV2AttentionMLA(
) )
if rope_scaling and rope_scaling.get("apply_yarn_scaling", True): if rope_scaling and rope_scaling.get("apply_yarn_scaling", True):
mscale_all_dim = rope_scaling.get("mscale_all_dim", False) self.scaling = compute_mla_mscale_scaling(rope_scaling, self.scaling)
scaling_factor = rope_scaling["factor"]
mscale = yarn_get_mscale(scaling_factor, float(mscale_all_dim))
self.scaling = self.scaling * mscale * mscale
else: else:
self.rotary_emb = None self.rotary_emb = None
self.use_deepseek_yarn_rope = rope_scaling is not None self.use_deepseek_yarn_rope = rope_scaling is not None