Update injection_main.py

injection_main.py

@@ -331,7 +331,12 @@ def invert(
     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
+        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)}")
+
         # We'll skip the final iteration
         if i >= num_inference_steps - 1:
             continue
@@ -378,6 +383,7 @@ def invert(
 
 
 
+
 def style_image_with_inversion(
     pipe,
     input_image,