Update block.py

block.py

@@ -14,15 +14,19 @@ from diffusers.schedulers import EulerDiscreteScheduler
 from diffusers.models import AutoencoderKL
 from diffusers.configuration_utils import FrozenDict
 
-from diffusers.modular_pipelines.stable_diffusion_xl.before_denoise import prepare_latents_img2img, StableDiffusionXLControlNetAutoInput
-from diffusers.modular_pipelines.stable_diffusion_xl.encoders import StableDiffusionXLAutoIPAdapterStep
+from diffusers.modular_pipelines.stable_diffusion_xl.before_denoise import prepare_latents_img2img
 from diffusers.modular_pipelines.stable_diffusion_xl.denoise import (
     StableDiffusionXLDenoiseLoopWrapper, 
     StableDiffusionXLLoopDenoiser, 
     StableDiffusionXLControlNetLoopDenoiser, 
     StableDiffusionXLLoopAfterDenoiser
 )
-from diffusers.modular_pipelines.stable_diffusion_xl import IMAGE2IMAGE_BLOCKS, TEXT2IMAGE_BLOCKS
+from diffusers.modular_pipelines.stable_diffusion_xl.modular_blocks_presets import(
+    IMAGE2IMAGE_BLOCKS, 
+    TEXT2IMAGE_BLOCK,
+    StableDiffusionXLAutoControlNetInputStep,
+    StableDiffusionXLAutoIPAdapterStep
+)
 
 import torch
 from typing import List, Tuple, Any, Optional