maybe

Author: hlky

src/diffusers/pipelines/flux/pipeline_flux_rfinversion_noise.py

@@ -627,6 +627,8 @@ def __call__(
         callback_on_step_end_tensor_inputs: List[str] = ["latents"],
         max_sequence_length: int = 512,
         sigmas = None,
+        flip_schedule = False,
+        even_timesteps = None,
     ):
         r"""
         Function invoked when calling the pipeline for generation.
@@ -781,6 +783,11 @@ def __call__(
             sigmas,
             mu=mu,
         )
+        if flip_schedule:
+            self.scheduler.sigmas = self.scheduler.sigmas.flip(0)
+            self.scheduler.timesteps = self.scheduler.timesteps.flip(0)
+        print(f"self.scheduler.sigmas {self.scheduler.sigmas}")
+        print(f"self.scheduler.timesteps {self.scheduler.timesteps}")
         timesteps, num_inference_steps = self.get_timesteps(num_inference_steps, strength, device)
 
         if num_inference_steps < 1:
@@ -827,8 +834,11 @@ def __call__(
                     continue
                 if self.interrupt:
                     continue
-                timestep = t.expand(latents.shape[0]).to(latents.dtype)
-                timestep = timestep / 1000
+                if even_timesteps is None:
+                    timestep = t.expand(latents.shape[0]).to(latents.dtype)
+                    timestep = timestep / 1000
+                else:
+                    timestep = even_timesteps[i]
                 noise_pred = self.transformer(
                     hidden_states=latents,
                     timestep=timestep,