Update injection_main.py

injection_main.py

@@ -324,14 +324,10 @@ def invert(
     # Set num inference steps
     pipe.scheduler.set_timesteps(num_inference_steps, device=device)
 
-    # Reversed timesteps
+    # Reversed timesteps <<<<<<<<<<<<<<<<<<<<
     timesteps = list(reversed(pipe.scheduler.timesteps))
 
-    for i in tqdm(
-        range(1, num_inference_steps),
-        total=num_inference_steps - 1,
-        desc="DDIM Inversion",
-    ):
+    for i in 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:
@@ -361,7 +357,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]