Revert "[Kernel] Expand MoE weight loading + Add Fused Marlin MoE Kernel (#7527)" (#7764)

This commit is contained in:
Michael Goin
2024-08-22 03:42:14 +00:00
committed by GitHub
parent cde9183b40
commit aae74ef95c
15 changed files with 84 additions and 2374 deletions
+2 -2
View File
@@ -23,11 +23,11 @@ def get_model_architecture(
architectures = getattr(model_config.hf_config, "architectures", [])
# Special handling for quantized Mixtral.
# FIXME(woosuk): This is a temporary hack.
mixtral_supported = ["fp8", "compressed-tensors"]
if (model_config.quantization is not None
and model_config.quantization not in mixtral_supported
and model_config.quantization != "fp8"
and "MixtralForCausalLM" in architectures):
architectures = ["QuantMixtralForCausalLM"]
return ModelRegistry.resolve_model_cls(architectures)