[CI] Fix precommit rope_theta issue (#29040)

Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
Wentao Ye
2025-11-19 17:22:43 -05:00
committed by GitHub
parent cb0a7b4bea
commit 0075bfffd4

View File

@@ -991,8 +991,7 @@ class DeepseekV2MLAAttention(nn.Module):
qk_rope_head_dim,
rotary_dim=qk_rope_head_dim,
max_position=max_position_embeddings,
base=rope_theta,
rope_scaling=rope_scaling,
rope_parameters=config.rope_parameters,
is_neox_style=True,
)
self.indexer = Indexer(