shorten loop subblock name

Author: yiyixuxu

src/diffusers/modular_pipelines/stable_diffusion_xl/denoise.py

@@ -42,7 +42,7 @@
 
 # YiYi experimenting composible denoise loop
 # loop step (1): prepare latent input for denoiser
-class StableDiffusionXLDenoiseLoopBeforeDenoiser(PipelineBlock):
+class StableDiffusionXLLoopBeforeDenoiser(PipelineBlock):
 
     model_name = "stable-diffusion-xl"
 
@@ -76,7 +76,7 @@ def __call__(self, components: StableDiffusionXLModularLoader, block_state: Bloc
         return components, block_state
 
 # loop step (1): prepare latent input for denoiser (with inpainting)
-class StableDiffusionXLInpaintDenoiseLoopBeforeDenoiser(PipelineBlock):
+class StableDiffusionXLInpaintLoopBeforeDenoiser(PipelineBlock):
 
     model_name = "stable-diffusion-xl"
 
@@ -147,7 +147,7 @@ def __call__(self, components: StableDiffusionXLModularLoader, block_state: Bloc
         return components, block_state
 
 # loop step (2): denoise the latents with guidance
-class StableDiffusionXLDenoiseLoopDenoiser(PipelineBlock):
+class StableDiffusionXLLoopDenoiser(PipelineBlock):
 
     model_name = "stable-diffusion-xl"
 
@@ -251,7 +251,7 @@ def __call__(self, components: StableDiffusionXLModularLoader, block_state: Bloc
         return components, block_state
 
 # loop step (2): denoise the latents with guidance (with controlnet)
-class StableDiffusionXLControlNetDenoiseLoopDenoiser(PipelineBlock):
+class StableDiffusionXLControlNetLoopDenoiser(PipelineBlock):
 
     model_name = "stable-diffusion-xl"
 
@@ -446,7 +446,7 @@ def __call__(self, components: StableDiffusionXLModularLoader, block_state: Bloc
         return components, block_state
 
 # loop step (3): scheduler step to update latents
-class StableDiffusionXLDenoiseLoopAfterDenoiser(PipelineBlock):
+class StableDiffusionXLLoopAfterDenoiser(PipelineBlock):
 
     model_name = "stable-diffusion-xl"
 
@@ -508,7 +508,7 @@ def __call__(self, components: StableDiffusionXLModularLoader, block_state: Bloc
         return components, block_state
 
 # loop step (3): scheduler step to update latents (with inpainting)
-class StableDiffusionXLInpaintDenoiseLoopAfterDenoiser(PipelineBlock):
+class StableDiffusionXLInpaintLoopAfterDenoiser(PipelineBlock):
 
     model_name = "stable-diffusion-xl"
 
@@ -680,7 +680,7 @@ def __call__(self, components: StableDiffusionXLModularLoader, state: PipelineSt
 
 # composing the denoising loops
 class StableDiffusionXLDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper):
-    block_classes = [StableDiffusionXLDenoiseLoopBeforeDenoiser, StableDiffusionXLDenoiseLoopDenoiser, StableDiffusionXLDenoiseLoopAfterDenoiser]
+    block_classes = [StableDiffusionXLLoopBeforeDenoiser, StableDiffusionXLLoopDenoiser, StableDiffusionXLLoopAfterDenoiser]
     block_names = ["before_denoiser", "denoiser", "after_denoiser"]
 
     @property
@@ -689,53 +689,53 @@ def description(self) -> str:
             "Denoise step that iteratively denoise the latents. \n"
             "Its loop logic is defined in `StableDiffusionXLDenoiseLoopWrapper.__call__` method \n"
             "At each iteration, it runs blocks defined in `blocks` sequencially:\n"
-            " - `StableDiffusionXLDenoiseLoopBeforeDenoiser`\n"
-            " - `StableDiffusionXLDenoiseLoopDenoiser`\n"
-            " - `StableDiffusionXLDenoiseLoopAfterDenoiser`\n"
+            " - `StableDiffusionXLLoopBeforeDenoiser`\n"
+            " - `StableDiffusionXLLoopDenoiser`\n"
+            " - `StableDiffusionXLLoopAfterDenoiser`\n"
         )
 
 # control_cond
 class StableDiffusionXLControlNetDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper):
-    block_classes = [StableDiffusionXLDenoiseLoopBeforeDenoiser, StableDiffusionXLControlNetDenoiseLoopDenoiser, StableDiffusionXLDenoiseLoopAfterDenoiser]
+    block_classes = [StableDiffusionXLLoopBeforeDenoiser, StableDiffusionXLControlNetLoopDenoiser, StableDiffusionXLLoopAfterDenoiser]
     block_names = ["before_denoiser", "denoiser", "after_denoiser"]
     @property
     def description(self) -> str:
         return (
             "Denoise step that iteratively denoise the latents with controlnet. \n"
             "Its loop logic is defined in  `StableDiffusionXLDenoiseLoopWrapper.__call__` method \n"
             "At each iteration, it runs blocks defined in `blocks` sequencially:\n"
-            " - `StableDiffusionXLDenoiseLoopBeforeDenoiser`\n"
-            " - `StableDiffusionXLControlNetDenoiseLoopDenoiser`\n"
-            " - `StableDiffusionXLDenoiseLoopAfterDenoiser`\n"
+            " - `StableDiffusionXLLoopBeforeDenoiser`\n"
+            " - `StableDiffusionXLControlNetLoopDenoiser`\n"
+            " - `StableDiffusionXLLoopAfterDenoiser`\n"
         )
 
 # mask
 class StableDiffusionXLInpaintDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper):
-    block_classes = [StableDiffusionXLInpaintDenoiseLoopBeforeDenoiser, StableDiffusionXLDenoiseLoopDenoiser, StableDiffusionXLInpaintDenoiseLoopAfterDenoiser]
+    block_classes = [StableDiffusionXLInpaintLoopBeforeDenoiser, StableDiffusionXLLoopDenoiser, StableDiffusionXLInpaintLoopAfterDenoiser]
     block_names = ["before_denoiser", "denoiser", "after_denoiser"]
     @property
     def description(self) -> str:
         return (
             "Denoise step that iteratively denoise the latents(for inpainting task only). \n"
             "Its loop logic is defined in `StableDiffusionXLDenoiseLoopWrapper.__call__` method \n"
             "At each iteration, it runs blocks defined in `blocks` sequencially:\n"
-            " - `StableDiffusionXLInpaintDenoiseLoopBeforeDenoiser`\n"
-            " - `StableDiffusionXLDenoiseLoopDenoiser`\n"
-            " - `StableDiffusionXLInpaintDenoiseLoopAfterDenoiser`\n"
+            " - `StableDiffusionXLInpaintLoopBeforeDenoiser`\n"
+            " - `StableDiffusionXLLoopDenoiser`\n"
+            " - `StableDiffusionXLInpaintLoopAfterDenoiser`\n"
         )
 # control_cond + mask
 class StableDiffusionXLInpaintControlNetDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper):
-    block_classes = [StableDiffusionXLInpaintDenoiseLoopBeforeDenoiser, StableDiffusionXLControlNetDenoiseLoopDenoiser, StableDiffusionXLInpaintDenoiseLoopAfterDenoiser]
+    block_classes = [StableDiffusionXLInpaintLoopBeforeDenoiser, StableDiffusionXLControlNetLoopDenoiser, StableDiffusionXLInpaintLoopAfterDenoiser]
     block_names = ["before_denoiser", "denoiser", "after_denoiser"]
     @property
     def description(self) -> str:
         return (
             "Denoise step that iteratively denoise the latents(for inpainting task only) with controlnet. \n"
             "Its loop logic is defined in `StableDiffusionXLDenoiseLoopWrapper.__call__` method \n"
             "At each iteration, it runs blocks defined in `blocks` sequencially:\n"
-            " - `StableDiffusionXLInpaintDenoiseLoopBeforeDenoiser`\n"
-            " - `StableDiffusionXLControlNetDenoiseLoopDenoiser`\n"
-            " - `StableDiffusionXLInpaintDenoiseLoopAfterDenoiser`\n"
+            " - `StableDiffusionXLInpaintLoopBeforeDenoiser`\n"
+            " - `StableDiffusionXLControlNetLoopDenoiser`\n"
+            " - `StableDiffusionXLInpaintLoopAfterDenoiser`\n"
         )