Skip to content

Commit 2bff2d5

Browse files
committed
try fix for tests
1 parent d080379 commit 2bff2d5

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/diffusers/hooks/group_offloading.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,7 @@ def __init__(
124124
group: ModuleGroup,
125125
next_group: Optional[ModuleGroup] = None,
126126
) -> None:
127+
super().__init__()
127128
self.group = group
128129
self.next_group = next_group
129130

@@ -168,6 +169,7 @@ class LazyPrefetchGroupOffloadingHook(ModelHook):
168169
_is_stateful = False
169170

170171
def __init__(self):
172+
super().__init__()
171173
self.execution_order: List[Tuple[str, torch.nn.Module]] = []
172174
self._layer_execution_tracker_module_names = set()
173175

@@ -253,6 +255,7 @@ class LayerExecutionTrackerHook(ModelHook):
253255
_is_stateful = False
254256

255257
def __init__(self, execution_order_update_callback):
258+
super().__init__()
256259
self.execution_order_update_callback = execution_order_update_callback
257260

258261
def pre_forward(self, module, *args, **kwargs):

0 commit comments

Comments
 (0)