Update block.py
Browse files
block.py
CHANGED
@@ -195,17 +195,17 @@ class SDXLDiffDiffLoopBeforeDenoiser(PipelineBlock):
|
|
195 |
return components, block_state
|
196 |
|
197 |
|
198 |
-
class
|
199 |
block_classes = [SDXLDiffDiffLoopBeforeDenoiser, StableDiffusionXLLoopDenoiser, StableDiffusionXLLoopAfterDenoiser]
|
200 |
block_names = ["before_denoiser", "denoiser", "after_denoiser"]
|
201 |
|
202 |
# control_cond
|
203 |
-
class
|
204 |
block_classes = [SDXLDiffDiffLoopBeforeDenoiser, StableDiffusionXLControlNetLoopDenoiser, StableDiffusionXLLoopAfterDenoiser]
|
205 |
block_names = ["before_denoiser", "denoiser", "after_denoiser"]
|
206 |
|
207 |
class SDXLDiffDiffAutoDenoiseStep(AutoPipelineBlocks):
|
208 |
-
block_classes = [
|
209 |
block_names = ["controlnet_denoise", "denoise"]
|
210 |
block_trigger_inputs = ["controlnet_cond", None]
|
211 |
|
|
|
195 |
return components, block_state
|
196 |
|
197 |
|
198 |
+
class SDXLDiffDiffDenoiseStep(StableDiffusionXLDenoiseLoopWrapper):
|
199 |
block_classes = [SDXLDiffDiffLoopBeforeDenoiser, StableDiffusionXLLoopDenoiser, StableDiffusionXLLoopAfterDenoiser]
|
200 |
block_names = ["before_denoiser", "denoiser", "after_denoiser"]
|
201 |
|
202 |
# control_cond
|
203 |
+
class SDXLDiffDiffControlNetDenoiseStep(StableDiffusionXLDenoiseLoopWrapper):
|
204 |
block_classes = [SDXLDiffDiffLoopBeforeDenoiser, StableDiffusionXLControlNetLoopDenoiser, StableDiffusionXLLoopAfterDenoiser]
|
205 |
block_names = ["before_denoiser", "denoiser", "after_denoiser"]
|
206 |
|
207 |
class SDXLDiffDiffAutoDenoiseStep(AutoPipelineBlocks):
|
208 |
+
block_classes = [SDXLDiffDiffControlNetDenoiseStep, SDXLDiffDiffDenoiseStep]
|
209 |
block_names = ["controlnet_denoise", "denoise"]
|
210 |
block_trigger_inputs = ["controlnet_cond", None]
|
211 |
|