Update block.py

block.py

@@ -21,7 +21,7 @@ from diffusers.modular_pipelines.stable_diffusion_xl.denoise import (
     StableDiffusionXLControlNetLoopDenoiser, 
     StableDiffusionXLLoopAfterDenoiser
 )
-from diffusers.modular_pipelines.stable_diffusion_xl.modular_blocks_presets import(
+from diffusers.modular_pipelines.stable_diffusion_xl.modular_blocks import(
     IMAGE2IMAGE_BLOCKS, 
     TEXT2IMAGE_BLOCKS,
     StableDiffusionXLAutoControlNetInputStep,
@@ -129,7 +129,7 @@ class SDXLDiffDiffPrepareLatentsStep(PipelineBlock):
         block_state.diffdiff_masks = diffdiff_map > (thresholds + (block_state.denoising_start or 0))
         block_state.original_latents = block_state.latents
 
-        self.add_block_state(state, block_state)
+        self.set_block_state(state, block_state)
 
         return components, state