Skip to content

Commit bb6412e

Browse files
Update vllm_ascend/ops/moe/token_dispatcher.py
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> Signed-off-by: Mercykid-bash <[email protected]>
1 parent 6102024 commit bb6412e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm_ascend/ops/moe/token_dispatcher.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def get_dispatch_mc2_kwargs(
123123
global_redundant_expert_num: int = 0,
124124
):
125125
quant_mode = 2 if self.with_quant else 0
126-
self.moe__expert_num = len(expert_map) + global_redundant_expert_num
126+
self.moe_expert_num = len(expert_map) + global_redundant_expert_num
127127
kwargs_mc2 = {
128128
"x": hidden_states,
129129
"expert_ids": topk_ids,

0 commit comments

Comments
 (0)