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

Update pipelines/pipeline_seesr.py

Browse files
Files changed (1) hide show
  1. pipelines/pipeline_seesr.py +4 -2
pipelines/pipeline_seesr.py CHANGED
@@ -1059,15 +1059,17 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
1059
  # Infer ControlNet only for the conditional batch.
1060
  controlnet_latent_model_input = latents
1061
  controlnet_prompt_embeds = prompt_embeds.chunk(2)[1]
 
1062
 
1063
  else:
1064
  controlnet_latent_model_input = latent_model_input
1065
  controlnet_prompt_embeds = prompt_embeds
 
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,
@@ -1152,7 +1154,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
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,
 
1059
  # Infer ControlNet only for the conditional batch.
1060
  controlnet_latent_model_input = latents
1061
  controlnet_prompt_embeds = prompt_embeds.chunk(2)[1]
1062
+ print("well unexpected")
1063
 
1064
  else:
1065
  controlnet_latent_model_input = latent_model_input
1066
  controlnet_prompt_embeds = prompt_embeds
1067
+ print("a possiblity")
1068
 
1069
  if h*w<=tile_size*tile_size: # tiled latent input
1070
  down_block_res_samples, mid_block_res_sample = [None]*10, None
1071
 
1072
+ print(f"controlnet 1 started with {controlnet_latent_model_input.shape}:{controlnet_prompt_embeds.shape}")
1073
 
1074
  down_block_res_samples, mid_block_res_sample = self.controlnet(
1075
  controlnet_latent_model_input,
 
1154
  cond_list_t = torch.cat(cond_list, dim=0)
1155
  img_list_t = torch.cat(img_list, dim=0)
1156
  #print(input_list_t.shape, cond_list_t.shape, img_list_t.shape, fg_mask_list_t.shape)
1157
+ print(f"controlnet 2 started with {cond_list_t.shape}:{controlnet_prompt_embeds.shape}")
1158
 
1159
  down_block_res_samples, mid_block_res_sample = self.controlnet(
1160
  cond_list_t,