Skip to content

Commit e7877e1

Browse files
a-r-r-o-wsayakpaul
andauthored
Update src/diffusers/hooks/group_offloading.py
Co-authored-by: Sayak Paul <[email protected]>
1 parent 234bf4d commit e7877e1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/hooks/group_offloading.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -314,7 +314,7 @@ def pre_forward(self, module: torch.nn.Module, *args, **kwargs):
314314
# If this group didn't onload itself, it means it was asynchronously onloaded by the
315315
# previous group. We need to synchronize the side stream to ensure parameters
316316
# are completely loaded to proceed with forward pass.
317-
# Also, we should only do this synchronize if we don't already do it from the sync call in
317+
# Also, we should only do this synchronization if we don't already do it from the sync call in
318318
# self.next_group.onload_, hence the `not should_onload_next_group` check.
319319
self.group.stream.synchronize()
320320

0 commit comments

Comments
 (0)