Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -63,9 +63,6 @@ latents = torch.randn(
|
|
| 63 |
latents = latents.to(torch_device)
|
| 64 |
latents = latents * scheduler.init_noise_sigma # Scaling (previous versions did latents = latents * self.scheduler.sigmas[0]
|
| 65 |
|
| 66 |
-
# Loop
|
| 67 |
-
with autocast("cuda"): # will fallback to CPU if no CUDA; no autocast for MPS
|
| 68 |
-
for i, t in tqdm(enumerate(scheduler.timesteps), total=len(scheduler.timesteps)):
|
| 69 |
|
| 70 |
|
| 71 |
|
|
|
|
| 63 |
latents = latents.to(torch_device)
|
| 64 |
latents = latents * scheduler.init_noise_sigma # Scaling (previous versions did latents = latents * self.scheduler.sigmas[0]
|
| 65 |
|
|
|
|
|
|
|
|
|
|
| 66 |
|
| 67 |
|
| 68 |
|