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 |
|