Corrected controlnet for unet types
Browse files- pipeline.py +2 -2
pipeline.py
CHANGED
|
@@ -1022,8 +1022,8 @@ class StableDiffusionControlNetImg2ImgPipeline(DiffusionPipeline, TextualInversi
|
|
| 1022 |
down_block_res_samples.append(down_block_res_sample)
|
| 1023 |
mid_block_res_samples.append(mid_block_res_sample)
|
| 1024 |
|
| 1025 |
-
down_block_res_samples =
|
| 1026 |
-
mid_block_res_sample = torch.cat(
|
| 1027 |
|
| 1028 |
# down_block_res_samples = torch.cat(down_block_res_samples, dim=0)
|
| 1029 |
# mid_block_res_sample = torch.cat(mid_block_res_samples, dim=0)
|
|
|
|
| 1022 |
down_block_res_samples.append(down_block_res_sample)
|
| 1023 |
mid_block_res_samples.append(mid_block_res_sample)
|
| 1024 |
|
| 1025 |
+
down_block_res_samples = tuple(down_block_res_samples)
|
| 1026 |
+
mid_block_res_sample = torch.cat(mid_block_res_samples, dim=0)
|
| 1027 |
|
| 1028 |
# down_block_res_samples = torch.cat(down_block_res_samples, dim=0)
|
| 1029 |
# mid_block_res_sample = torch.cat(mid_block_res_samples, dim=0)
|