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