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