[Bugfix] Handle DeepseekV4ForCausalLM in benchmark_moe get_model_params (#52044)
Co-authored-by: Do_it_now_! <23432123@users.noreply.github.com>
This commit is contained in:
co-authored by
Do_it_now_!
parent
e0e5a7fb28
commit
101c4477dd
@@ -766,6 +766,7 @@ def get_model_params(config):
|
||||
"DeepseekV2ForCausalLM",
|
||||
"DeepseekV3ForCausalLM",
|
||||
"DeepseekV32ForCausalLM",
|
||||
"DeepseekV4ForCausalLM",
|
||||
"GlmMoeDsaForCausalLM",
|
||||
"Glm4MoeForCausalLM",
|
||||
"Glm4MoeLiteForCausalLM",
|
||||
|
||||
Reference in New Issue
Block a user