Spaces:
Running
on
A10G
Running
on
A10G
Linoy Tsaban
commited on
Commit
•
7992e19
1
Parent(s):
e4ae53e
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)) if prog_bar else reversed(timesteps)
|
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)]
|