Skip to content

Commit acbe140

Browse files
committed
Fix use_lu_lambdas and use_karras_sigmas with beta_schedule=squaredcos_cap_v2 in DPMSolverMultistepScheduler
1 parent cd0a4a8 commit acbe140

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/schedulers/scheduling_dpmsolver_multistep.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -399,12 +399,12 @@ def set_timesteps(
399399
if self.config.use_karras_sigmas:
400400
sigmas = np.flip(sigmas).copy()
401401
sigmas = self._convert_to_karras(in_sigmas=sigmas, num_inference_steps=num_inference_steps)
402-
timesteps = np.array([self._sigma_to_t(sigma, log_sigmas) for sigma in sigmas]).round()
402+
timesteps = np.array([self._sigma_to_t(sigma, log_sigmas) for sigma in sigmas])
403403
elif self.config.use_lu_lambdas:
404404
lambdas = np.flip(log_sigmas.copy())
405405
lambdas = self._convert_to_lu(in_lambdas=lambdas, num_inference_steps=num_inference_steps)
406406
sigmas = np.exp(lambdas)
407-
timesteps = np.array([self._sigma_to_t(sigma, log_sigmas) for sigma in sigmas]).round()
407+
timesteps = np.array([self._sigma_to_t(sigma, log_sigmas) for sigma in sigmas])
408408
elif self.config.use_exponential_sigmas:
409409
sigmas = np.flip(sigmas).copy()
410410
sigmas = self._convert_to_exponential(in_sigmas=sigmas, num_inference_steps=num_inference_steps)

0 commit comments

Comments
 (0)