Update app.py
Browse files
app.py
CHANGED
|
@@ -1062,7 +1062,7 @@ text_embeddings = torch.cat([uncond_embeddings, text_embeddings])
|
|
| 1062 |
|
| 1063 |
# Prep Scheduler
|
| 1064 |
set_timesteps(scheduler, num_inference_steps)
|
| 1065 |
-
|
| 1066 |
# Prep latents
|
| 1067 |
latents = torch.randn(
|
| 1068 |
(batch_size, unet.in_channels, height // 8, width // 8),
|
|
|
|
| 1062 |
|
| 1063 |
# Prep Scheduler
|
| 1064 |
set_timesteps(scheduler, num_inference_steps)
|
| 1065 |
+
|
| 1066 |
# Prep latents
|
| 1067 |
latents = torch.randn(
|
| 1068 |
(batch_size, unet.in_channels, height // 8, width // 8),
|