make style

Author: a-r-r-o-w

src/diffusers/hooks/faster_cache.py

@@ -33,7 +33,7 @@
 _SPATIAL_ATTENTION_BLOCK_IDENTIFIERS = (
     "^blocks.*attn",
     "^transformer_blocks.*attn",
-    "^single_transformer_blocks.*attn"
+    "^single_transformer_blocks.*attn",
 )
 _TEMPORAL_ATTENTION_BLOCK_IDENTIFIERS = ("^temporal_transformer_blocks.*attn",)
 _TRANSFORMER_BLOCK_IDENTIFIERS = _SPATIAL_ATTENTION_BLOCK_IDENTIFIERS + _TEMPORAL_ATTENTION_BLOCK_IDENTIFIERS
@@ -483,10 +483,7 @@ def reset_state(self, module: torch.nn.Module) -> torch.nn.Module:
         return module
 
 
-def apply_faster_cache(
-    module: torch.nn.Module,
-    config: FasterCacheConfig
-) -> None:
+def apply_faster_cache(module: torch.nn.Module, config: FasterCacheConfig) -> None:
     r"""
     Applies [FasterCache](https://huggingface.co/papers/2410.19355) to a given pipeline.
 

src/diffusers/hooks/pyramid_attention_broadcast.py

@@ -175,10 +175,7 @@ def reset_state(self, module: torch.nn.Module) -> None:
         return module
 
 
-def apply_pyramid_attention_broadcast(
-    module: torch.nn.Module,
-    config: PyramidAttentionBroadcastConfig
-):
+def apply_pyramid_attention_broadcast(module: torch.nn.Module, config: PyramidAttentionBroadcastConfig):
     r"""
     Apply [Pyramid Attention Broadcast](https://huggingface.co/papers/2408.12588) to a given pipeline.
 

tests/pipelines/test_pipelines_common.py

@@ -2609,7 +2609,6 @@ def faster_cache_state_check_callback(pipe, i, t, kwargs):
             if not hasattr(module, "_diffusers_hook"):
                 continue
 
-
             if name == "":
                 # Root denoiser module
                 state = module._diffusers_hook.get_hook(_FASTER_CACHE_DENOISER_HOOK).state