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

Update pipelines/pipeline_seesr.py

Browse files
Files changed (1) hide show
  1. pipelines/pipeline_seesr.py +4 -0
pipelines/pipeline_seesr.py CHANGED
@@ -1066,6 +1066,9 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
1066
 
1067
  if h*w<=tile_size*tile_size: # tiled latent input
1068
  down_block_res_samples, mid_block_res_sample = [None]*10, None
 
 
 
1069
  down_block_res_samples, mid_block_res_sample = self.controlnet(
1070
  controlnet_latent_model_input,
1071
  t,
@@ -1149,6 +1152,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
1149
  cond_list_t = torch.cat(cond_list, dim=0)
1150
  img_list_t = torch.cat(img_list, dim=0)
1151
  #print(input_list_t.shape, cond_list_t.shape, img_list_t.shape, fg_mask_list_t.shape)
 
1152
 
1153
  down_block_res_samples, mid_block_res_sample = self.controlnet(
1154
  cond_list_t,
 
1066
 
1067
  if h*w<=tile_size*tile_size: # tiled latent input
1068
  down_block_res_samples, mid_block_res_sample = [None]*10, None
1069
+
1070
+ print(f"controlnet started with {controlnet_latent_model_input.shape}:{controlnet_prompt_embeds.shape}")
1071
+
1072
  down_block_res_samples, mid_block_res_sample = self.controlnet(
1073
  controlnet_latent_model_input,
1074
  t,
 
1152
  cond_list_t = torch.cat(cond_list, dim=0)
1153
  img_list_t = torch.cat(img_list, dim=0)
1154
  #print(input_list_t.shape, cond_list_t.shape, img_list_t.shape, fg_mask_list_t.shape)
1155
+ print(f"controlnet started with {cond_list_t.shape}:{controlnet_prompt_embeds.shape}")
1156
 
1157
  down_block_res_samples, mid_block_res_sample = self.controlnet(
1158
  cond_list_t,