Skip to content

Commit d4abb86

Browse files
authored
[None][fix] fix EPLB for DeepSeek-V3.2-Exp (#9245)
Signed-off-by: Fanrong Li <[email protected]>
1 parent f6ec6e2 commit d4abb86

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tensorrt_llm/_torch/modules/fused_moe/moe_load_balancer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1072,6 +1072,7 @@ def __exit__(self, exc_type, exc_val, exc_tb):
10721072

10731073
moe_model_arch_list = [
10741074
'DeepseekV3ForCausalLM',
1075+
'DeepseekV32ForCausalLM',
10751076
'GptOssForCausalLM',
10761077
'MixtralForCausalLM',
10771078
'Llama4ForConditionalGeneration',

0 commit comments

Comments
 (0)