Spaces:
Runtime error
Runtime error
Linoy Tsaban
commited on
Commit
β’
f1baf70
1
Parent(s):
ca0ffc2
Update inversion_utils.py
Browse files- 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 |
-
|
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]:])}
|