MrAlex commited on
Commit
90804b7
·
1 Parent(s): d898808

Corrected controlnet for unet types

Browse files
Files changed (1) hide show
  1. 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 = torch.cat([torch.tensor(x).unsqueeze(0) for x in down_block_res_samples], dim=0)
1026
- mid_block_res_sample = torch.cat([torch.tensor(x).unsqueeze(0) for x in 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)
 
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)