YiYiXu
/

modular-diffdiff-0704

Model card Files Files and versions

YiYiXu commited on Jul 5

Commit

57fe4d0

·

verified ·

1 Parent(s): 821a600

Update block.py

Files changed (1) hide show

block.py +1 -1

block.py CHANGED Viewed

@@ -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)
         return components, state

         block_state.diffdiff_masks = diffdiff_map > (thresholds + (block_state.denoising_start or 0))
         block_state.original_latents = block_state.latents
+        self.set_block_state(state, block_state)
         return components, state