Spaces:
Runtime error
Runtime error
piyushgrover
commited on
Commit
Β·
abcb468
1
Parent(s):
2ef7d9b
Update utils.py
Browse files
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]
|