Skip to content

Commit b72b4b5

Browse files
committed
revise the sigmas of DPMSolverMultistepInverseScheduler for flow_prediction
1 parent 9b721db commit b72b4b5

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/diffusers/schedulers/scheduling_dpmsolver_multistep_inverse.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -299,7 +299,9 @@ def set_timesteps(self, num_inference_steps: int = None, device: Union[str, torc
299299
elif self.config.use_flow_sigmas:
300300
alphas = np.linspace(1, 1 / self.config.num_train_timesteps, num_inference_steps + 1)
301301
sigmas = 1.0 - alphas
302-
sigmas = np.flip(self.config.flow_shift * sigmas / (1 + (self.config.flow_shift - 1) * sigmas))[:-1].copy()
302+
sigmas = np.flip(self.config.flow_shift * sigmas / (1 + (self.config.flow_shift - 1) * sigmas))[:-1][
303+
::-1
304+
].copy()
303305
timesteps = (sigmas * self.config.num_train_timesteps).copy()
304306
sigmas = np.concatenate([sigmas, sigmas[-1:]]).astype(np.float32)
305307
else:

0 commit comments

Comments
 (0)