Skip to content

Commit a9a24fb

Browse files
authored
fix builder for kimi-k2 (#4069)
1 parent adac4b1 commit a9a24fb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lmdeploy/pytorch/configurations/deepseek_v2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ class DeepseekV2ModelConfigBuilder(AutoModelConfigBuilder):
1010
@classmethod
1111
def condition(cls, hf_config):
1212
"""config."""
13-
return hf_config.model_type in ['deepseek_v3', 'deepseek_v2']
13+
return hf_config.model_type in ['deepseek_v3', 'deepseek_v2', 'kimi_k2']
1414

1515
@classmethod
1616
def build(cls, hf_config, model_path: str = None, **kwargs):

0 commit comments

Comments
 (0)