From 2d280f173f3e937facf18820fe1db472bd67e6f7 Mon Sep 17 00:00:00 2001 From: YiYi Xu Date: Mon, 21 Oct 2024 13:27:01 -1000 Subject: [PATCH] fix singlestep dpm tests (#9716) fix Co-authored-by: Sayak Paul --- src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py b/src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py index 3329919cfb..6841a34a64 100644 --- a/src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py +++ b/src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py @@ -330,6 +330,7 @@ class DPMSolverSinglestepScheduler(SchedulerMixin, ConfigMixin): # Clipping the minimum of all lambda(t) for numerical stability. # This is critical for cosine (squaredcos_cap_v2) noise schedule. clipped_idx = torch.searchsorted(torch.flip(self.lambda_t, [0]), self.config.lambda_min_clipped) + clipped_idx = clipped_idx.item() timesteps = ( np.linspace(0, self.config.num_train_timesteps - 1 - clipped_idx, num_inference_steps + 1) .round()[::-1][:-1]