From f4df2af9465a32b4ac2a42e58eba04a1188c224c Mon Sep 17 00:00:00 2001 From: ilmarkov Date: Mon, 24 Nov 2025 18:58:46 +0000 Subject: [PATCH] Wip Signed-off-by: ilmarkov --- vllm/distributed/eplb/rebalance_execute.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/distributed/eplb/rebalance_execute.py b/vllm/distributed/eplb/rebalance_execute.py index 376dad8a72ef1..046f3895970c4 100644 --- a/vllm/distributed/eplb/rebalance_execute.py +++ b/vllm/distributed/eplb/rebalance_execute.py @@ -436,7 +436,7 @@ def rearrange_expert_weights_inplace( is_unchanged=is_unchanged, is_received_locally=is_received_locally, experts_recv_loc=experts_recv_loc, - new_indices=new_global_expert_indices[layer].tolist(), + new_indices=new_global_expert_indices_cpu[layer].tolist(), ep_group=ep_group, )