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