make style

Author: a-r-r-o-w

src/diffusers/hooks/hooks.py

@@ -134,7 +134,9 @@ def new_forward(module, *args, **kwargs):
                 output = old_forward(*args, **kwargs)
                 return hook.post_forward(module, output)
 
-        self._module_ref.forward = functools.update_wrapper(functools.partial(new_forward, self._module_ref), old_forward)
+        self._module_ref.forward = functools.update_wrapper(
+            functools.partial(new_forward, self._module_ref), old_forward
+        )
 
         self.hooks[name] = hook
         self._hook_order.append(name)
@@ -156,7 +158,7 @@ def reset_stateful_hooks(self, recurse: bool = True) -> None:
             hook = self.hooks[hook_name]
             if hook._is_stateful:
                 hook.reset_state(self._module_ref)
-        
+
         if recurse:
             for module in self._module_ref.modules():
                 if hasattr(module, "_diffusers_hook"):

src/diffusers/hooks/pyramid_attention_broadcast.py

@@ -106,7 +106,7 @@ def __init__(self) -> None:
     def reset(self):
         self.iteration = 0
         self.cache = None
-    
+
     def __repr__(self):
         cache_repr = ""
         if self.cache is None:
@@ -251,7 +251,7 @@ def _apply_pyramid_attention_broadcast_on_attention_class(
     def skip_callback(module: torch.nn.Module) -> bool:
         hook: PyramidAttentionBroadcastHook = module._diffusers_hook.get_hook("pyramid_attention_broadcast")
         pab_state: PyramidAttentionBroadcastState = hook.state
-        
+
         if pab_state.cache is None:
             return False