YiYiXu commited on
Commit
ba73c5b
·
verified ·
1 Parent(s): cda6908

Update block.py

Browse files
Files changed (1) hide show
  1. block.py +3 -3
block.py CHANGED
@@ -195,17 +195,17 @@ class SDXLDiffDiffLoopBeforeDenoiser(PipelineBlock):
195
  return components, block_state
196
 
197
 
198
- class SDXLDiffDiffDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper):
199
  block_classes = [SDXLDiffDiffLoopBeforeDenoiser, StableDiffusionXLLoopDenoiser, StableDiffusionXLLoopAfterDenoiser]
200
  block_names = ["before_denoiser", "denoiser", "after_denoiser"]
201
 
202
  # control_cond
203
- class SDXLDiffDiffControlNetDenoiseLoop(StableDiffusionXLDenoiseLoopWrapper):
204
  block_classes = [SDXLDiffDiffLoopBeforeDenoiser, StableDiffusionXLControlNetLoopDenoiser, StableDiffusionXLLoopAfterDenoiser]
205
  block_names = ["before_denoiser", "denoiser", "after_denoiser"]
206
 
207
  class SDXLDiffDiffAutoDenoiseStep(AutoPipelineBlocks):
208
- block_classes = [SDXLDiffDiffControlNetDenoiseLoop, SDXLDiffDiffDenoiseLoop]
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