Spaces:
Running
on
A10G
Running
on
A10G
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)]
|