piyushgrover commited on
Commit
abcb468
Β·
1 Parent(s): 2ef7d9b

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -1
utils.py CHANGED
@@ -161,7 +161,8 @@ def generate_with_embs(text_input, text_embeddings, output=None, generator=None,
161
  latents = latents * scheduler.init_noise_sigma
162
 
163
  # Loop
164
- for i, t in tqdm(enumerate(scheduler.timesteps), total=len(scheduler.timesteps)):
 
165
  # expand the latents if we are doing classifier-free guidance to avoid doing two forward passes.
166
  latent_model_input = torch.cat([latents] * 2)
167
  sigma = scheduler.sigmas[i]
 
161
  latents = latents * scheduler.init_noise_sigma
162
 
163
  # Loop
164
+ #for i, t in tqdm(enumerate(scheduler.timesteps), total=len(scheduler.timesteps)):
165
+ for i, t in enumerate(scheduler.timesteps):
166
  # expand the latents if we are doing classifier-free guidance to avoid doing two forward passes.
167
  latent_model_input = torch.cat([latents] * 2)
168
  sigma = scheduler.sigmas[i]