mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-10 07:15:01 +08:00
[BugFix] 'DeepseekV2Config' object has no attribute 'use_mla'` (#28387)
Signed-off-by: Lin, Fanli <fanli.lin@intel.com>
This commit is contained in:
parent
f0359fffa4
commit
c7991269dd
@ -456,7 +456,11 @@ class KimiVLForConditionalGeneration(nn.Module, SupportsMultiModal, SupportsPP):
|
||||
(".gate_up_proj", ".gate_proj", 0),
|
||||
(".gate_up_proj", ".up_proj", 1),
|
||||
]
|
||||
if not config.use_mla:
|
||||
use_mha = (
|
||||
config.model_type == "deepseek"
|
||||
or config.qk_nope_head_dim + config.qk_rope_head_dim == 0
|
||||
)
|
||||
if use_mha:
|
||||
stacked_params_mapping += [
|
||||
(".qkv_proj", ".q_proj", "q"),
|
||||
(".qkv_proj", ".k_proj", "k"),
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user