Spaces:
Runtime error
Runtime error
Linoy Tsaban
commited on
Commit
·
0aa753f
1
Parent(s):
40001af
Update inversion_utils.py
Browse files- inversion_utils.py +1 -1
inversion_utils.py
CHANGED
|
@@ -157,7 +157,7 @@ def inversion_forward_process(model, x0,
|
|
| 157 |
|
| 158 |
t_to_idx = {int(v):k for k,v in enumerate(timesteps)}
|
| 159 |
xt = x0
|
| 160 |
-
op =
|
| 161 |
|
| 162 |
for t in op:
|
| 163 |
idx = t_to_idx[int(t)]
|
|
|
|
| 157 |
|
| 158 |
t_to_idx = {int(v):k for k,v in enumerate(timesteps)}
|
| 159 |
xt = x0
|
| 160 |
+
op = tqdm(reversed(timesteps)) if prog_bar else reversed(timesteps)
|
| 161 |
|
| 162 |
for t in op:
|
| 163 |
idx = t_to_idx[int(t)]
|