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