Spaces:

fffiloni
/

Artist

Running on Zero

fffiloni commited on Jul 29

Commit

aa39831

•

1 Parent(s): 5a109bc

Update injection_main.py

Files changed (1) hide show

injection_main.py CHANGED Viewed

@@ -327,7 +327,7 @@ def invert(
     # Reversed timesteps <<<<<<<<<<<<<<<<<<<<
     timesteps = reversed(pipe.scheduler.timesteps)
-    for i in progress.tqdm(
         range(1, num_inference_steps),
         total=num_inference_steps - 1,
         desc="DDIM Inversion",

     # Reversed timesteps <<<<<<<<<<<<<<<<<<<<
     timesteps = reversed(pipe.scheduler.timesteps)
+    for i in Progress.tqdm(
         range(1, num_inference_steps),
         total=num_inference_steps - 1,
         desc="DDIM Inversion",