Upload pipeline.py
Browse files- 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 |
)
|