Update injection_main.py

injection_main.py

@@ -325,15 +325,9 @@ def invert(
     pipe.scheduler.set_timesteps(num_inference_steps, device=device)
 
     # Reversed timesteps <<<<<<<<<<<<<<<<<<<<
-    timesteps = reversed(pipe.scheduler.timesteps)
+    timesteps = list(reversed(pipe.scheduler.timesteps))
 
-    for i in Progress.tqdm(
-        range(1, num_inference_steps),
-        total=num_inference_steps - 1,
-        desc="DDIM Inversion",
-    ):
-
-        # We'll skip the final iteration
+    for i in range(num_inference_steps):
         if i >= num_inference_steps - 1:
             continue
 
@@ -361,7 +355,7 @@ def invert(
             )
 
         current_t = max(0, t.item() - (1000 // num_inference_steps))  # t
-        next_t = t  # min(999, t.item() + (1000//num_inference_steps)) # t+1
+        next_t = t  # min(999, t.item() + (1000 // num_inference_steps)) # t+1
         alpha_t = pipe.scheduler.alphas_cumprod[current_t]
         alpha_t_next = pipe.scheduler.alphas_cumprod[next_t]