Update injection_main.py

injection_main.py

@@ -324,8 +324,8 @@ def invert(
     # Set num inference steps
     pipe.scheduler.set_timesteps(num_inference_steps, device=device)
 
-    # Reversed timesteps <<<<<<<<<<<<<<<<<<<<
-    timesteps = reversed(pipe.scheduler.timesteps)
+    # Reversed timesteps
+    timesteps = list(reversed(pipe.scheduler.timesteps))
 
     for i in tqdm(
         range(1, num_inference_steps),
@@ -361,7 +361,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]
 
@@ -376,6 +376,7 @@ def invert(
     return torch.cat(intermediate_latents)
 
 
+
 def style_image_with_inversion(
     pipe,
     input_image,