Linoy Tsaban
commited on
Commit
·
5589998
1
Parent(s):
b005a25
Update inversion_utils.py
Browse files- inversion_utils.py +1 -1
inversion_utils.py
CHANGED
|
@@ -262,7 +262,7 @@ def inversion_reverse_process(model,
|
|
| 262 |
timesteps = model.scheduler.timesteps.to(model.device)
|
| 263 |
|
| 264 |
xt = xT.expand(batch_size, -1, -1, -1)
|
| 265 |
-
op =
|
| 266 |
|
| 267 |
t_to_idx = {int(v):k for k,v in enumerate(timesteps[-zs.shape[0]:])}
|
| 268 |
|
|
|
|
| 262 |
timesteps = model.scheduler.timesteps.to(model.device)
|
| 263 |
|
| 264 |
xt = xT.expand(batch_size, -1, -1, -1)
|
| 265 |
+
op = tqdm(timesteps[-zs.shape[0]:]) if prog_bar else timesteps[-zs.shape[0]:]
|
| 266 |
|
| 267 |
t_to_idx = {int(v):k for k,v in enumerate(timesteps[-zs.shape[0]:])}
|
| 268 |
|