alexnasa commited on
Commit
9554532
·
verified ·
1 Parent(s): c577ed3

Update pipelines/pipeline_seesr.py

Browse files
Files changed (1) hide show
  1. pipelines/pipeline_seesr.py +1 -0
pipelines/pipeline_seesr.py CHANGED
@@ -1169,6 +1169,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
1169
  mid_block_res_sample = torch.cat([torch.zeros_like(mid_block_res_sample), mid_block_res_sample])
1170
 
1171
  # predict the noise residual
 
1172
  model_out = self.unet(
1173
  input_list_t,
1174
  t,
 
1169
  mid_block_res_sample = torch.cat([torch.zeros_like(mid_block_res_sample), mid_block_res_sample])
1170
 
1171
  # predict the noise residual
1172
+ print(f"unet started with {input_list_t.shape}:{prompt_embeds.shape}")
1173
  model_out = self.unet(
1174
  input_list_t,
1175
  t,