empathy

Author: sayakpaul

src/diffusers/modular_pipelines/flux/modular_blocks.py

@@ -87,18 +87,8 @@ def description(self):
 
 # text2image
 class FluxAutoBlocks(SequentialPipelineBlocks):
-    block_classes = [
-        FluxTextEncoderStep,
-        FluxAutoBeforeDenoiseStep,
-        FluxAutoDenoiseStep,
-        FluxAutoDecodeStep,
-    ]
-    block_names = [
-        "text_encoder",
-        "before_denoise",
-        "denoise",
-        "decoder",
-    ]
+    block_classes = [FluxTextEncoderStep, FluxAutoBeforeDenoiseStep, FluxAutoDenoiseStep, FluxAutoDecodeStep]
+    block_names = ["text_encoder", "before_denoise", "denoise", "decoder"]
 
     @property
     def description(self):
@@ -132,7 +122,4 @@ def description(self):
 )
 
 
-ALL_BLOCKS = {
-    "text2image": TEXT2IMAGE_BLOCKS,
-    "auto": AUTO_BLOCKS,
-}
+ALL_BLOCKS = {"text2image": TEXT2IMAGE_BLOCKS, "auto": AUTO_BLOCKS}