mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-15 13:55:38 +08:00
fix#30092 Kimi-Linear model loading failure with missing indexer_rotary_emb (#30093)
Signed-off-by: baonudesifeizhai <baonudesifeizhai@gmail.com>
This commit is contained in:
parent
c4d62618ca
commit
e3fbb6f152
@ -24,9 +24,9 @@ class MLAModules:
|
|||||||
q_b_proj: torch.nn.Module | None
|
q_b_proj: torch.nn.Module | None
|
||||||
q_proj: torch.nn.Module | None
|
q_proj: torch.nn.Module | None
|
||||||
indexer: torch.nn.Module | None
|
indexer: torch.nn.Module | None
|
||||||
indexer_rotary_emb: torch.nn.Module | None
|
|
||||||
is_sparse: bool
|
is_sparse: bool
|
||||||
topk_indices_buffer: torch.Tensor | None
|
topk_indices_buffer: torch.Tensor | None
|
||||||
|
indexer_rotary_emb: torch.nn.Module | None = None
|
||||||
|
|
||||||
|
|
||||||
@CustomOp.register("multi_head_latent_attention")
|
@CustomOp.register("multi_head_latent_attention")
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user