Spaces:
Running
on
Zero
Running
on
Zero
Update pipelines/pipeline_seesr.py
Browse files
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,
|