Linoy Tsaban commited on
Commit
f1baf70
1 Parent(s): ca0ffc2

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
  assert len(etas) == model.scheduler.num_inference_steps
263
  timesteps = model.scheduler.timesteps.to(model.device)
264
 
265
- # xt = xT.expand(batch_size, -1, -1, -1)
266
  op = tqdm(timesteps[-zs.shape[0]:]) if prog_bar else timesteps[-zs.shape[0]:]
267
 
268
  t_to_idx = {int(v):k for k,v in enumerate(timesteps[-zs.shape[0]:])}
 
262
  assert len(etas) == model.scheduler.num_inference_steps
263
  timesteps = model.scheduler.timesteps.to(model.device)
264
 
265
+ xt = xT.expand(batch_size, -1, -1, -1)
266
  op = tqdm(timesteps[-zs.shape[0]:]) if prog_bar else timesteps[-zs.shape[0]:]
267
 
268
  t_to_idx = {int(v):k for k,v in enumerate(timesteps[-zs.shape[0]:])}