Spaces:
Running
on
Zero
Running
on
Zero
Update pipelines/pipeline_seesr.py
Browse files
pipelines/pipeline_seesr.py
CHANGED
@@ -1169,6 +1169,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
|
|
1169 |
mid_block_res_sample = torch.cat([torch.zeros_like(mid_block_res_sample), mid_block_res_sample])
|
1170 |
|
1171 |
# predict the noise residual
|
|
|
1172 |
model_out = self.unet(
|
1173 |
input_list_t,
|
1174 |
t,
|
|
|
1169 |
mid_block_res_sample = torch.cat([torch.zeros_like(mid_block_res_sample), mid_block_res_sample])
|
1170 |
|
1171 |
# predict the noise residual
|
1172 |
+
print(f"unet started with {input_list_t.shape}:{prompt_embeds.shape}")
|
1173 |
model_out = self.unet(
|
1174 |
input_list_t,
|
1175 |
t,
|