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

Update pipelines/pipeline_seesr.py

Browse files
Files changed (1) hide show
  1. pipelines/pipeline_seesr.py +1 -0
pipelines/pipeline_seesr.py CHANGED
@@ -1086,6 +1086,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
1086
  mid_block_res_sample = torch.cat([torch.zeros_like(mid_block_res_sample), mid_block_res_sample])
1087
 
1088
  # predict the noise residual
 
1089
  noise_pred = self.unet(
1090
  latent_model_input,
1091
  t,
 
1086
  mid_block_res_sample = torch.cat([torch.zeros_like(mid_block_res_sample), mid_block_res_sample])
1087
 
1088
  # predict the noise residual
1089
+ print(f"unet started with {latent_model_input.shape}:{prompt_embeds.shape}")
1090
  noise_pred = self.unet(
1091
  latent_model_input,
1092
  t,