Spaces:
Runtime error
Runtime error
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]
|