Update block.py

block.py

@@ -17,9 +17,9 @@ from diffusers.configuration_utils import FrozenDict
 from diffusers.modular_pipelines.stable_diffusion_xl.before_denoise import prepare_latents_img2img
 from diffusers.modular_pipelines.stable_diffusion_xl.denoise import (
     StableDiffusionXLDenoiseLoopWrapper, 
-    StableDiffusionXLDenoiseLoopDenoiser, 
-    StableDiffusionXLControlNetDenoiseLoopDenoiser, 
-    StableDiffusionXLDenoiseLoopAfterDenoiser
+    StableDiffusionXLLoopDenoiser, 
+    StableDiffusionXLControlNetLoopDenoiser, 
+    StableDiffusionXLLoopAfterDenoiser
 )
 from diffusers.modular_pipelines.stable_diffusion_xl.modular_pipeline_block_mappings import (
     IMAGE2IMAGE_BLOCKS, 
@@ -132,7 +132,7 @@ class SDXLDiffDiffPrepareLatentsStep(PipelineBlock):
         return components, state
 
 
-class SDXLDiffDiffDenoiseLoopBeforeDenoiser(PipelineBlock):
+class SDXLDiffDiffLoopBeforeDenoiser(PipelineBlock):
     model_name = "stable-diffusion-xl"
 
     @property
@@ -194,12 +194,12 @@ class SDXLDiffDiffDenoiseLoopBeforeDenoiser(PipelineBlock):
 
 
 class SDXLDiffDiffDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper):
-    block_classes = [SDXLDiffDiffDenoiseLoopBeforeDenoiser, StableDiffusionXLDenoiseLoopDenoiser, StableDiffusionXLDenoiseLoopAfterDenoiser]
+    block_classes = [SDXLDiffDiffLoopBeforeDenoiser, StableDiffusionXLLoopDenoiser, StableDiffusionXLLoopAfterDenoiser]
     block_names = ["before_denoiser", "denoiser", "after_denoiser"]
 
 # control_cond
 class SDXLDiffDiffControlNetDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper):
-    block_classes = [SDXLDiffDiffDenoiseLoopBeforeDenoiser, StableDiffusionXLControlNetDenoiseLoopDenoiser, StableDiffusionXLDenoiseLoopAfterDenoiser]
+    block_classes = [SDXLDiffDiffLoopBeforeDenoiser, StableDiffusionXLControlNetLoopDenoiser, StableDiffusionXLLoopAfterDenoiser]
     block_names = ["before_denoiser", "denoiser", "after_denoiser"]
 
 class SDXLDiffDiffDenoiseStep(AutoPipelineBlocks):