Spaces:
Running
Running
Linoy Tsaban
commited on
Commit
•
c78a44c
1
Parent(s):
7992e19
Update inversion_utils.py
Browse files- inversion_utils.py +1 -1
inversion_utils.py
CHANGED
@@ -155,7 +155,7 @@ def inversion_forward_process(model, x0,
|
|
155 |
|
156 |
t_to_idx = {int(v):k for k,v in enumerate(timesteps)}
|
157 |
xt = x0
|
158 |
-
op = tqdm(reversed(timesteps), desc=
|
159 |
|
160 |
for t in op:
|
161 |
idx = t_to_idx[int(t)]
|
|
|
155 |
|
156 |
t_to_idx = {int(v):k for k,v in enumerate(timesteps)}
|
157 |
xt = x0
|
158 |
+
op = tqdm(reversed(timesteps), desc= "Inverting...") if prog_bar else reversed(timesteps)
|
159 |
|
160 |
for t in op:
|
161 |
idx = t_to_idx[int(t)]
|