fix(benchmark): support Glm4MoeLite in fused MoE tuner (#37623)
This commit is contained in:
@@ -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",
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user