Spaces:
Runtime error
Runtime error
Linoy Tsaban
commited on
Commit
•
5589998
1
Parent(s):
b005a25
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 |
timesteps = model.scheduler.timesteps.to(model.device)
|
263 |
|
264 |
xt = xT.expand(batch_size, -1, -1, -1)
|
265 |
-
op =
|
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 |
|