Skip to content

Commit 533bc5d

Browse files
author
yexin
committed
fix rebase issues
1 parent 5266ac1 commit 533bc5d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

checkpoint_engine/worker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ def update_weights_from_ipc(self, zmq_handles: dict[str, str]):
168168

169169
update_weights_from_ipc(
170170
self._zmq_ctx,
171-
zmq_handles[device_uuid],
171+
zmq_handles[self._device_uuid],
172172
device_id=self.device.index,
173173
run=self.model_runner.model.load_weights,
174174
post_hook=lambda: process_weights_after_loading(

0 commit comments

Comments
 (0)