Linoy Tsaban commited on
Commit
0aa753f
1 Parent(s): 40001af

Update inversion_utils.py

Browse files
Files changed (1) hide show
  1. 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 = progress.tqdm(reversed(timesteps)) if prog_bar else reversed(timesteps)
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)]