fix(benchmark): support Glm4MoeLite in fused MoE tuner (#37623)

This commit is contained in:
Xiaoyu Zhang
2026-09-03 17:21:42 +08:00
committed by GitHub
parent 7ed29eba80
commit 397aeca376
@@ -96,6 +96,7 @@ def get_model_config(
"DeepseekV32ForCausalLM",
"DeepseekV4ForCausalLM",
"Glm4MoeForCausalLM",
"Glm4MoeLiteForCausalLM",
"GlmMoeDsaForCausalLM",
"KimiVLForConditionalGeneration",
"MistralLarge3ForCausalLM",
@@ -108,6 +109,7 @@ def get_model_config(
"DeepseekV3ForCausalLM",
"DeepseekV32ForCausalLM",
"Glm4MoeForCausalLM",
"Glm4MoeLiteForCausalLM",
"GlmMoeDsaForCausalLM",
"MistralLarge3ForCausalLM",
]