Skip to content

Commit 3ef2172

Browse files
replace mixtral
1 parent e1ce33a commit 3ef2172

File tree

3 files changed

+254
-252
lines changed

3 files changed

+254
-252
lines changed

vllm/model_executor/models/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,9 +36,9 @@
3636
# For decapoda-research/llama-*
3737
"LLaMAForCausalLM": ("llama", "LlamaForCausalLM"),
3838
"MistralForCausalLM": ("llama", "LlamaForCausalLM"),
39-
"MixtralForCausalLM": ("mixtral", "MixtralForCausalLM"),
39+
"MixtralForCausalLM": ("mixtral", "MixtralForCausalLM"), ##
4040
"QuantMixtralForCausalLM": ("mixtral_quant", "MixtralForCausalLM"),
41-
"PhiMoEForCausalLM": ("phimoe", "PhiMoEForCausalLM"),
41+
#"PhiMoEForCausalLM": ("phimoe", "PhiMoEForCausalLM"),
4242
# transformers's mpt class has lower case
4343
"MptForCausalLM": ("mpt", "MPTForCausalLM"),
4444
"MPTForCausalLM": ("mpt", "MPTForCausalLM"),

0 commit comments

Comments
 (0)