Update block.py

block.py

@@ -199,19 +199,20 @@ class SDXLDiffDiffControlNetDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper):
     block_classes = [SDXLDiffDiffLoopBeforeDenoiser, StableDiffusionXLControlNetLoopDenoiser, StableDiffusionXLLoopAfterDenoiser]
     block_names = ["before_denoiser", "denoiser", "after_denoiser"]
 
-class SDXLDiffDiffDenoiseStep(AutoPipelineBlocks):
+class SDXLDiffDiffAutoDenoiseStep(AutoPipelineBlocks):
     block_classes = [SDXLDiffDiffControlNetDenoiseLoop, SDXLDiffDiffDenoiseLoop]
     block_names = ["controlnet_denoise", "denoise"]
     block_trigger_inputs = ["controlnet_cond", None]
 
 
-DIFFDIFF_BLOCKS = IMAGE2IMAGE_BLOCKS.copy()
-DIFFDIFF_BLOCKS["denoise"] = SDXLDiffDiffDenoiseStep
-DIFFDIFF_BLOCKS["prepare_latents"] = SDXLDiffDiffPrepareLatentsStep
-DIFFDIFF_BLOCKS["set_timesteps"] = TEXT2IMAGE_BLOCKS["set_timesteps"]
-
+DIFFDIFF_AUTO_BLOCKS = IMAGE2IMAGE_BLOCKS.copy()
+DIFFDIFF_AUTO_BLOCKS["prepare_latents"] = SDXLDiffDiffPrepareLatentsStep
+DIFFDIFF_AUTO_BLOCKS["set_timesteps"] = TEXT2IMAGE_BLOCKS["set_timesteps"]
+DIFFDIFF_AUTO_BLOCKS["denoise"] = SDXLDiffDiffAutoDenoiseStep
+DIFFDIFF_AUTO_BLOCKS.insert("ip_adapter", StableDiffusionXLAutoIPAdapterStep, 0)
+DIFFDIFF_AUTO_BLOCKS.insert("controlnet_input",StableDiffusionXLControlNetAutoInput, 7)
 
 class DiffDiffBlocks(SequentialPipelineBlocks):
-    block_classes = list(DIFFDIFF_BLOCKS.values())
-    block_names = list(DIFFDIFF_BLOCKS.keys())
+    block_classes = list(DIFFDIFF_AUTO_BLOCKS.values())
+    block_names = list(DIFFDIFF_AUTO_BLOCKS.keys())