Skip to content

Commit 1c2a3c7

Browse files
authored
Update scheduling_dpmsolver_multistep.py
1 parent c438ad4 commit 1c2a3c7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/schedulers/scheduling_dpmsolver_multistep.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -407,7 +407,7 @@ def set_timesteps(
407407
lambdas = self._convert_to_lu(in_lambdas=lambdas, num_inference_steps=num_inference_steps)
408408
sigmas = np.exp(lambdas)
409409
timesteps = np.array([self._sigma_to_t(sigma, log_sigmas) for sigma in sigmas])
410-
if self.config.beta_schedule in {"linear", "scaled_linear"}:
410+
if self.config.beta_schedule != "squaredcos_cap_v2":
411411
timesteps = timesteps.round()
412412
elif self.config.use_exponential_sigmas:
413413
sigmas = np.flip(sigmas).copy()

0 commit comments

Comments
 (0)