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,
|