From 9a41b9130e37c362217a3bcf6e04709fd7b6eb63 Mon Sep 17 00:00:00 2001 From: ilmarkov Date: Fri, 12 Dec 2025 14:52:49 +0000 Subject: [PATCH] Fix precommit Signed-off-by: ilmarkov --- vllm/distributed/eplb/eplb_state.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/distributed/eplb/eplb_state.py b/vllm/distributed/eplb/eplb_state.py index 111479f2ee2f4..961544290fed5 100644 --- a/vllm/distributed/eplb/eplb_state.py +++ b/vllm/distributed/eplb/eplb_state.py @@ -1005,7 +1005,7 @@ class EplbState: is_received_locally=model_state.is_received_locally, recv_metadata=model_state.recv_metadata, new_indices=new_indices, - ep_group=ep_group, + ep_rank=ep_group.rank(), ) transferred_layer = model_state.layer_to_transfer self._update_layer_mapping_from_new(model_state, transferred_layer)