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