Remove naive rotary forward overriding. (#19263)

This commit is contained in:
Jue Wang
2026-03-04 00:50:40 -05:00
committed by GitHub
parent ac1f07487a
commit 5972f97f11

View File

@@ -1195,8 +1195,6 @@ class DeepseekV2AttentionMLA(
scaling_factor = rope_scaling["factor"]
mscale = yarn_get_mscale(scaling_factor, float(mscale_all_dim))
self.scaling = self.scaling * mscale * mscale
else:
self.rotary_emb.forward = self.rotary_emb.forward_native
else:
self.rotary_emb = None
self.use_deepseek_yarn_rope = rope_scaling is not None