Update injection_main.py

injection_main.py

@@ -325,16 +325,13 @@ def invert(
     pipe.scheduler.set_timesteps(num_inference_steps, device=device)
 
     # Reversed timesteps <<<<<<<<<<<<<<<<<<<<
-    timesteps = list(reversed(pipe.scheduler.timesteps))
+    timesteps = reversed(pipe.scheduler.timesteps)
 
-    # Initialize tqdm progress bar
-    for i in range(num_inference_steps - 1):
-        print(f"Iteration type: {type(i)}, value: {i}")  # Debugging statement
-
-        if isinstance(i, (int, float)):
-            i = int(i)
-        else:
-            raise TypeError(f"Unexpected type for iteration variable: {type(i)}")
+    for i in progress.tqdm(
+        range(1, num_inference_steps),
+        total=num_inference_steps - 1,
+        desc="DDIM Inversion",
+    ):
 
         # We'll skip the final iteration
         if i >= num_inference_steps - 1: