[Kimi-Linear] Correct prefixes and add compatibility to AWQ quants (vllm-project#27834)

Signed-off-by: toncao <[email protected]>
Co-authored-by: toncao <[email protected]>

Author: toncao

vllm/model_executor/models/kimi_linear.py

@@ -155,6 +155,7 @@ def __init__(
                 hidden_act=config.hidden_act,
                 quant_config=quant_config,
                 reduce_results=False,
+                prefix=f"{prefix}.shared_experts",
             )
 
     def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
@@ -340,7 +341,7 @@ def __init__(
             self.block_sparse_moe = KimiMoE(
                 config=config,
                 quant_config=quant_config,
-                prefix=f"{prefix}.mlp",
+                prefix=f"{prefix}.block_sparse_moe",
             )
             self.mlp = self.block_sparse_moe
         else: