AYYasaswini commited on
Commit
5c54b87
·
verified ·
1 Parent(s): fd9c32c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
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