Spaces:
Running
on
Zero
Running
on
Zero
Update pipelines/pipeline_seesr.py
Browse files
pipelines/pipeline_seesr.py
CHANGED
|
@@ -1086,6 +1086,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
|
|
| 1086 |
mid_block_res_sample = torch.cat([torch.zeros_like(mid_block_res_sample), mid_block_res_sample])
|
| 1087 |
|
| 1088 |
# predict the noise residual
|
|
|
|
| 1089 |
noise_pred = self.unet(
|
| 1090 |
latent_model_input,
|
| 1091 |
t,
|
|
|
|
| 1086 |
mid_block_res_sample = torch.cat([torch.zeros_like(mid_block_res_sample), mid_block_res_sample])
|
| 1087 |
|
| 1088 |
# predict the noise residual
|
| 1089 |
+
print(f"unet started with {latent_model_input.shape}:{prompt_embeds.shape}")
|
| 1090 |
noise_pred = self.unet(
|
| 1091 |
latent_model_input,
|
| 1092 |
t,
|