Skip to content

Commit 744064d

Browse files
princepride0xrushi
authored andcommitted
Adjusted the model order of the model registration file (vllm-project#26798)
Signed-off-by: 汪志鹏 <wangzhipeng628@gmail.com> Signed-off-by: 0xrushi <6279035+0xrushi@users.noreply.github.com>
1 parent eee5da2 commit 744064d

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

vllm/model_executor/models/registry.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -60,9 +60,6 @@
6060
"AquilaForCausalLM": ("llama", "LlamaForCausalLM"), # AquilaChat2
6161
"ArceeForCausalLM": ("arcee", "ArceeForCausalLM"),
6262
"ArcticForCausalLM": ("arctic", "ArcticForCausalLM"),
63-
"MiniMaxForCausalLM": ("minimax_text_01", "MiniMaxText01ForCausalLM"),
64-
"MiniMaxText01ForCausalLM": ("minimax_text_01", "MiniMaxText01ForCausalLM"),
65-
"MiniMaxM1ForCausalLM": ("minimax_text_01", "MiniMaxText01ForCausalLM"),
6663
# baichuan-7b, upper case 'C' in the class name
6764
"BaiChuanForCausalLM": ("baichuan", "BaiChuanForCausalLM"),
6865
# baichuan-13b, lower case 'c' in the class name
@@ -87,8 +84,10 @@
8784
"Ernie4_5_MoeForCausalLM": ("ernie45_moe", "Ernie4_5_MoeForCausalLM"),
8885
"ExaoneForCausalLM": ("exaone", "ExaoneForCausalLM"),
8986
"Exaone4ForCausalLM": ("exaone4", "Exaone4ForCausalLM"),
90-
"FalconForCausalLM": ("falcon", "FalconForCausalLM"),
9187
"Fairseq2LlamaForCausalLM": ("fairseq2_llama", "Fairseq2LlamaForCausalLM"),
88+
"FalconForCausalLM": ("falcon", "FalconForCausalLM"),
89+
"FalconMambaForCausalLM": ("mamba", "MambaForCausalLM"),
90+
"FalconH1ForCausalLM": ("falcon_h1", "FalconH1ForCausalLM"),
9291
"FlexOlmoForCausalLM": ("flex_olmo", "FlexOlmoForCausalLM"),
9392
"GemmaForCausalLM": ("gemma", "GemmaForCausalLM"),
9493
"Gemma2ForCausalLM": ("gemma2", "Gemma2ForCausalLM"),
@@ -126,11 +125,12 @@
126125
"LLaMAForCausalLM": ("llama", "LlamaForCausalLM"),
127126
"LongcatFlashForCausalLM": ("longcat_flash", "LongcatFlashForCausalLM"),
128127
"MambaForCausalLM": ("mamba", "MambaForCausalLM"),
129-
"FalconMambaForCausalLM": ("mamba", "MambaForCausalLM"),
130-
"FalconH1ForCausalLM": ("falcon_h1", "FalconH1ForCausalLM"),
131128
"Mamba2ForCausalLM": ("mamba2", "Mamba2ForCausalLM"),
132129
"MiniCPMForCausalLM": ("minicpm", "MiniCPMForCausalLM"),
133130
"MiniCPM3ForCausalLM": ("minicpm3", "MiniCPM3ForCausalLM"),
131+
"MiniMaxForCausalLM": ("minimax_text_01", "MiniMaxText01ForCausalLM"),
132+
"MiniMaxText01ForCausalLM": ("minimax_text_01", "MiniMaxText01ForCausalLM"),
133+
"MiniMaxM1ForCausalLM": ("minimax_text_01", "MiniMaxText01ForCausalLM"),
134134
"MistralForCausalLM": ("llama", "LlamaForCausalLM"),
135135
"MixtralForCausalLM": ("mixtral", "MixtralForCausalLM"),
136136
# transformers's mpt class has lower case

0 commit comments

Comments
 (0)