after_denoise -> decoders

Author: yiyixuxu

src/diffusers/modular_pipelines/stable_diffusion_xl/__init__.py

@@ -24,7 +24,7 @@
     _import_structure["modular_pipeline_presets"] = ["StableDiffusionXLAutoPipeline"]
     _import_structure["modular_loader"] = ["StableDiffusionXLModularLoader"]
     _import_structure["encoders"] = ["StableDiffusionXLAutoIPAdapterStep", "StableDiffusionXLTextEncoderStep", "StableDiffusionXLAutoVaeEncoderStep"]
-    _import_structure["after_denoise"] = ["StableDiffusionXLAutoDecodeStep"]
+    _import_structure["decoders"] = ["StableDiffusionXLAutoDecodeStep"]
 
 if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
     try:
@@ -36,7 +36,7 @@
         from .modular_pipeline_presets import StableDiffusionXLAutoPipeline
         from .modular_loader import StableDiffusionXLModularLoader
         from .encoders import StableDiffusionXLAutoIPAdapterStep, StableDiffusionXLTextEncoderStep, StableDiffusionXLAutoVaeEncoderStep
-        from .after_denoise import StableDiffusionXLAutoDecodeStep
+        from .decoders import StableDiffusionXLAutoDecodeStep
 else:
     import sys
 

src/diffusers/modular_pipelines/stable_diffusion_xl/after_denoise.py renamed to src/diffusers/modular_pipelines/stable_diffusion_xl/decoders.py

@@ -18,15 +18,15 @@
 
 from .denoise import StableDiffusionXLAutoDenoiseStep
 from .before_denoise import StableDiffusionXLAutoBeforeDenoiseStep
-from .after_denoise import StableDiffusionXLAutoDecodeStep
+from .decoders import StableDiffusionXLAutoDecodeStep
 from .encoders import StableDiffusionXLTextEncoderStep, StableDiffusionXLAutoIPAdapterStep, StableDiffusionXLAutoVaeEncoderStep
 
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name
 
 
 class StableDiffusionXLAutoPipeline(SequentialPipelineBlocks):
     block_classes = [StableDiffusionXLTextEncoderStep, StableDiffusionXLAutoIPAdapterStep, StableDiffusionXLAutoVaeEncoderStep, StableDiffusionXLAutoBeforeDenoiseStep, StableDiffusionXLAutoDenoiseStep, StableDiffusionXLAutoDecodeStep]
-    block_names = ["text_encoder", "ip_adapter", "image_encoder", "before_denoise", "denoise", "after_denoise"]
+    block_names = ["text_encoder", "ip_adapter", "image_encoder", "before_denoise", "denoise", "decoder"]
 
     @property
     def description(self):