[Bug Fix] Fix MM mtp incorrect rope emb (#6581)

* [Bug Fix] Fix MM mtp incorrect rope emb
This commit is contained in:
ming1753
2026-03-03 19:28:59 +08:00
committed by GitHub
parent 4ff3f4212f
commit c5eb6b65e7
3 changed files with 56 additions and 19 deletions
@@ -140,8 +140,6 @@ class AppendAttentionBackend(AttentionBackend):
self.rope_3d: bool = getattr(fd_config.model_config, "rope_3d", False) or getattr(
fd_config.model_config, "use_3d_rope", False
)
if fd_config.speculative_config.model_type != "main":
self.rope_3d = False
self.causal: bool = getattr(fd_config.model_config, "causal", True)
self.speculative_method: str = fd_config.speculative_config.method
self.speculate_max_draft_token_num: int = fd_config.speculative_config.num_speculative_tokens