Fix precommit

Signed-off-by: ilmarkov <markovilya197@gmail.com>
This commit is contained in:
ilmarkov 2025-12-12 14:52:49 +00:00
parent 040ae89c5e
commit 9a41b9130e

View File

@ -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)