codeysun commited on
Commit
125d9fa
·
verified ·
1 Parent(s): 2514258

Upload pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -0
pipeline.py CHANGED
@@ -1514,6 +1514,8 @@ class StableDiffusionControlNetPipeline(
1514
  masked_down_block_res_samples = (
1515
  masked_down_block_res_samples + (down_block_res_sample,)
1516
  )
 
 
1517
  mid_block_res_sample = self.apply_effective_region_mask(
1518
  effective_region_mask, mid_block_res_sample
1519
  )
 
1514
  masked_down_block_res_samples = (
1515
  masked_down_block_res_samples + (down_block_res_sample,)
1516
  )
1517
+ down_block_res_samples = masked_down_block_res_samples
1518
+
1519
  mid_block_res_sample = self.apply_effective_region_mask(
1520
  effective_region_mask, mid_block_res_sample
1521
  )