mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-13 18:55:24 +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", ".gate_proj", 0),
|
||||||
(".gate_up_proj", ".up_proj", 1),
|
(".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 += [
|
stacked_params_mapping += [
|
||||||
(".qkv_proj", ".q_proj", "q"),
|
(".qkv_proj", ".q_proj", "q"),
|
||||||
(".qkv_proj", ".k_proj", "k"),
|
(".qkv_proj", ".k_proj", "k"),
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user