diff --git a/vllm/distributed/eplb/async_worker.py b/vllm/distributed/eplb/async_worker.py index 6897015375614..3e4017e86b300 100644 --- a/vllm/distributed/eplb/async_worker.py +++ b/vllm/distributed/eplb/async_worker.py @@ -130,7 +130,7 @@ async def transfer_run_periodically( rebalancing_algorithm_executed = True layer_idx = model_state.layer_to_transfer - old_layer_indices = model_state.old_physical_to_logical_map[ + old_layer_indices = model_state.physical_to_logical_map[ layer_idx ] new_layer_indices = model_state.new_physical_to_logical_map[