From b72b4b5c9e1db392538b7bc49465191ef88d9ce5 Mon Sep 17 00:00:00 2001 From: bhzhang Date: Sun, 31 Aug 2025 00:16:23 +0800 Subject: [PATCH] revise the sigmas of DPMSolverMultistepInverseScheduler for flow_prediction --- .../schedulers/scheduling_dpmsolver_multistep_inverse.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/diffusers/schedulers/scheduling_dpmsolver_multistep_inverse.py b/src/diffusers/schedulers/scheduling_dpmsolver_multistep_inverse.py index 9ec958851111..54c884540b05 100644 --- a/src/diffusers/schedulers/scheduling_dpmsolver_multistep_inverse.py +++ b/src/diffusers/schedulers/scheduling_dpmsolver_multistep_inverse.py @@ -299,7 +299,9 @@ def set_timesteps(self, num_inference_steps: int = None, device: Union[str, torc elif self.config.use_flow_sigmas: alphas = np.linspace(1, 1 / self.config.num_train_timesteps, num_inference_steps + 1) sigmas = 1.0 - alphas - sigmas = np.flip(self.config.flow_shift * sigmas / (1 + (self.config.flow_shift - 1) * sigmas))[:-1].copy() + sigmas = np.flip(self.config.flow_shift * sigmas / (1 + (self.config.flow_shift - 1) * sigmas))[:-1][ + ::-1 + ].copy() timesteps = (sigmas * self.config.num_train_timesteps).copy() sigmas = np.concatenate([sigmas, sigmas[-1:]]).astype(np.float32) else: