We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c438ad4 commit 1c2a3c7Copy full SHA for 1c2a3c7
src/diffusers/schedulers/scheduling_dpmsolver_multistep.py
@@ -407,7 +407,7 @@ def set_timesteps(
407
lambdas = self._convert_to_lu(in_lambdas=lambdas, num_inference_steps=num_inference_steps)
408
sigmas = np.exp(lambdas)
409
timesteps = np.array([self._sigma_to_t(sigma, log_sigmas) for sigma in sigmas])
410
- if self.config.beta_schedule in {"linear", "scaled_linear"}:
+ if self.config.beta_schedule != "squaredcos_cap_v2":
411
timesteps = timesteps.round()
412
elif self.config.use_exponential_sigmas:
413
sigmas = np.flip(sigmas).copy()
0 commit comments