Skip to content

Commit cce88a5

Browse files
author
offline0806
committed
[BugFix]Fix expert map.
Signed-off-by: offline0806 <z00858301@china.huawei.com>
1 parent 627bbdd commit cce88a5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm_ascend/ops/common_fused_moe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -333,7 +333,7 @@ def forward_impl(self, hidden_states: torch.Tensor,
333333
renormalize=self.renormalize,
334334
use_grouped_topk=self.use_grouped_topk,
335335
global_num_experts=self.global_num_experts,
336-
# expert_map=self.expert_map,
336+
expert_map=self.expert_map,
337337
topk_group=self.topk_group,
338338
num_expert_group=self.num_expert_group,
339339
custom_routing_function=self.custom_routing_function,

0 commit comments

Comments
 (0)