Skip to content

Commit 6f7e859

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 40b6a91 commit 6f7e859

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

applications/ColossalChat/coati/distributed/zero_bubble/distributor.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,10 @@ def loop(self):
6565
self.profiler.exit(f"sync_model_consumer_pp_{i}")
6666
self.weight_version[i] += 1
6767
if all(
68-
[signal.get(f"producer_{self.distributor_id}_pp_{i}", None) == "ready_sync_model" for i in range(self.consumer_pp_size)]
68+
[
69+
signal.get(f"producer_{self.distributor_id}_pp_{i}", None) == "ready_sync_model"
70+
for i in range(self.consumer_pp_size)
71+
]
6972
):
7073
for i in range(self.consumer_pp_size):
7174
self.profiler.enter(f"sync_model_producer_{self.distributor_id}_pp_{i}")
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
11
ray==2.49.2
2-
pygloo>=0.2.0 # you need to build from source: https://github.com/ray-project/pygloo commit 82ae2d72222aefcac54a8e88995735ede3abe9cf https://github.com/ray-project/pygloo/blob/main/README.md
2+
pygloo>=0.2.0 # you need to build from source: https://github.com/ray-project/pygloo commit 82ae2d72222aefcac54a8e88995735ede3abe9cf https://github.com/ray-project/pygloo/blob/main/README.md

0 commit comments

Comments
 (0)