Fix kimi-linear launch server error (#21752)
Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
This commit is contained in:
@@ -491,6 +491,11 @@ class ModelConfig:
|
|||||||
self.qk_rope_head_dim = self.hf_config.qk_rope_head_dim
|
self.qk_rope_head_dim = self.hf_config.qk_rope_head_dim
|
||||||
self.v_head_dim = self.hf_config.v_head_dim
|
self.v_head_dim = self.hf_config.v_head_dim
|
||||||
self.qk_nope_head_dim = self.hf_config.qk_nope_head_dim
|
self.qk_nope_head_dim = self.hf_config.qk_nope_head_dim
|
||||||
|
self.scaling = 1 / math.sqrt(self.qk_nope_head_dim + self.qk_rope_head_dim)
|
||||||
|
if self.hf_config.rope_scaling:
|
||||||
|
self.scaling = compute_mla_mscale_scaling(
|
||||||
|
self.hf_config.rope_scaling, self.scaling
|
||||||
|
)
|
||||||
elif (
|
elif (
|
||||||
"BailingMoeV2_5ForCausalLM" in self.hf_config.architectures
|
"BailingMoeV2_5ForCausalLM" in self.hf_config.architectures
|
||||||
or "BailingMoeForCausalLMNextN" in self.hf_config.architectures
|
or "BailingMoeForCausalLMNextN" in self.hf_config.architectures
|
||||||
|
|||||||
Reference in New Issue
Block a user