[Bugfix] Register axk1 config to fix A.X-K1 init (#49727)
Signed-off-by: Djordje Ramic <djoramic@amd.com>
djramic committed
Jul 24, 2026 at 21:57 UTC
e222c33f2f568b54b41d2ebfc44f6c85ea072737
3 files changed
+4
-3
vllm/transformers_utils/config.py
+1
@@ -72,6 +72,7 @@ class LazyConfigDict(dict):
72
_CONFIG_REGISTRY: dict[str, type[PretrainedConfig]] = LazyConfigDict(
73
afmoe="AfmoeConfig",
74
arctic="ArcticConfig",
75
+ axk1="AXK1Config",
76
bagel="BagelConfig",
77
umm="CheersConfig",
78
chatglm="ChatGLMConfig",
vllm/transformers_utils/configs/AXK1.py
+1
-1
@@ -114,7 +114,7 @@ class AXK1Config(PretrainedConfig):
114
The dropout ratio for the attention probabilities.
115
"""
116
117
- model_type = "AXK1"
117
+ model_type = "axk1"
118
keys_to_ignore_at_inference = ["past_key_values"]
119
120
def __init__(
vllm/transformers_utils/model_arch_config_convertor.py
+2
-2
@@ -261,7 +261,7 @@ class ModelArchConfigConvertorBase:
261
if not hasattr(self.hf_text_config, "model_type"):
262
return False
263
elif self.hf_text_config.model_type in (
264
- "AXK1",
264
+ "axk1",
265
"deepseek_v2",
266
"deepseek_v3",
267
"deepseek_v32",
@@ -290,7 +290,7 @@ class ModelArchConfigConvertorBase:
290
return (
291
self.hf_text_config.model.model_type
292
in (
293
- "AXK1",
293
+ "axk1",
294
"deepseek_v2",
295
"deepseek_v3",
296
"deepseek_v32",