Unverified Commit 2d280f17 authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

fix singlestep dpm tests (#9716)



fix
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 63a0c9e5
......@@ -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]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment