Linoy Tsaban commited on
Commit
5589998
1 Parent(s): b005a25

Update inversion_utils.py

Browse files
Files changed (1) hide show
  1. inversion_utils.py +1 -1
inversion_utils.py CHANGED
@@ -262,7 +262,7 @@ def inversion_reverse_process(model,
262
  timesteps = model.scheduler.timesteps.to(model.device)
263
 
264
  xt = xT.expand(batch_size, -1, -1, -1)
265
- op = progress.tqdm(timesteps[-zs.shape[0]:]) if prog_bar else timesteps[-zs.shape[0]:]
266
 
267
  t_to_idx = {int(v):k for k,v in enumerate(timesteps[-zs.shape[0]:])}
268
 
 
262
  timesteps = model.scheduler.timesteps.to(model.device)
263
 
264
  xt = xT.expand(batch_size, -1, -1, -1)
265
+ op = tqdm(timesteps[-zs.shape[0]:]) if prog_bar else timesteps[-zs.shape[0]:]
266
 
267
  t_to_idx = {int(v):k for k,v in enumerate(timesteps[-zs.shape[0]:])}
268