From 65788e46edfb60a31782a2bda0ba01f594359785 Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Wed, 29 Jun 2022 15:12:58 +0200 Subject: [PATCH] add scaled_linear schedule in DDIM --- src/diffusers/schedulers/scheduling_ddim.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/diffusers/schedulers/scheduling_ddim.py b/src/diffusers/schedulers/scheduling_ddim.py index f626cb1ca5..70fe2490e4 100644 --- a/src/diffusers/schedulers/scheduling_ddim.py +++ b/src/diffusers/schedulers/scheduling_ddim.py @@ -71,6 +71,9 @@ class DDIMScheduler(SchedulerMixin, ConfigMixin): if beta_schedule == "linear": self.betas = np.linspace(beta_start, beta_end, timesteps, dtype=np.float32) + elif beta_schedule == "scaled_linear": + # this schedule is very specific to the latent diffusion model. + self.betas = np.linspace(beta_start**0.5, beta_end**0.5, timesteps, dtype=np.float32) ** 2 elif beta_schedule == "squaredcos_cap_v2": # Glide cosine schedule self.betas = betas_for_alpha_bar(timesteps)