MrAlex commited on
Commit
d898808
·
1 Parent(s): 25620ce

another fix

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([x.unsqueeze(0) for x in down_block_res_samples], dim=0)
1026
- mid_block_res_sample = torch.cat([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 = 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)