fix

a-r-r-o-w · a-r-r-o-w · commit 278366999fd0 · 2025-01-09T02:37:18.000+01:00
diff --git a/src/diffusers/hooks/group_offloading.py b/src/diffusers/hooks/group_offloading.py
@@ -134,7 +134,7 @@ def _apply_group_offloading_diffusers_block(
             group = ModuleGroup(
                 blocks, offload_device, onload_device, offload_leader=blocks[-1], onload_leader=blocks[0]
             )
-            should_offload = force_offload or i == 0
+            should_offload = force_offload or i > 0
             _apply_group_offloading(group, should_offload)
 
 

Original file line number	Diff line number	Diff line change
`@@ -134,7 +134,7 @@ def _apply_group_offloading_diffusers_block(`
`134`	`134`	`group = ModuleGroup(`
`135`	`135`	`blocks, offload_device, onload_device, offload_leader=blocks[-1], onload_leader=blocks[0]`
`136`	`136`	`)`
`137`		`- should_offload = force_offload or i == 0`
	`137`	`+ should_offload = force_offload or i > 0`
`138`	`138`	`_apply_group_offloading(group, should_offload)`
`139`	`139`
`140`	`140`