Skip to content

Commit 7656b63

Browse files
committed
fix qwen3-30b-a3b-base missing shape for to_huggingface conversion
1 parent 7ac8f66 commit 7656b63

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

src/maxtext/checkpoint_conversion/utils/hf_shape.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1131,6 +1131,7 @@ def MIXTRAL_HF_WEIGHTS_TO_SHAPE(config):
11311131
"llama3.1-70b": LLAMA31_HF_WEIGHTS_TO_SHAPE,
11321132
"llama3.1-405b": LLAMA31_HF_WEIGHTS_TO_SHAPE,
11331133
"qwen3-30b-a3b": QWEN_HF_WEIGHTS_TO_SHAPE,
1134+
"qwen3-30b-a3b-base": QWEN_HF_WEIGHTS_TO_SHAPE,
11341135
"qwen3-235b-a22b": QWEN_HF_WEIGHTS_TO_SHAPE,
11351136
"qwen3-480b-a35b": QWEN_HF_WEIGHTS_TO_SHAPE,
11361137
"deepseek2-16b": DEEPSEEK_HF_WEIGHTS_TO_SHAPE,

0 commit comments

Comments
 (0)