Update injection_main.py

injection_main.py

@@ -328,9 +328,10 @@ def invert(
     timesteps = list(reversed(pipe.scheduler.timesteps))
 
     # Initialize tqdm progress bar
-    progress_bar = tqdm(range(1, num_inference_steps), total=num_inference_steps - 1, desc="DDIM Inversion")
+    progress_bar = tqdm(range(num_inference_steps - 1), total=num_inference_steps - 1, desc="DDIM Inversion")
 
     for i in progress_bar:
+        i = int(i)  # Explicitly convert i to int
         # We'll skip the final iteration
         if i >= num_inference_steps - 1:
             continue
@@ -376,6 +377,7 @@ def invert(
 
 
 
+
 def style_image_with_inversion(
     pipe,
     input_image,