From 36a710ccd362e3d2c9ef8c9d87ba2caaabecfbf5 Mon Sep 17 00:00:00 2001 From: Jacob Helwig <60412857+JacobHelwig@users.noreply.github.com> Date: Fri, 24 Jan 2025 17:25:06 -0600 Subject: [PATCH] Sigmoid scheduler in scheduling_ddpm.py docs --- src/diffusers/schedulers/scheduling_ddpm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/schedulers/scheduling_ddpm.py b/src/diffusers/schedulers/scheduling_ddpm.py index eb40d79b9f60..624d5a5cd4f3 100644 --- a/src/diffusers/schedulers/scheduling_ddpm.py +++ b/src/diffusers/schedulers/scheduling_ddpm.py @@ -142,7 +142,7 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin): The final `beta` value. beta_schedule (`str`, defaults to `"linear"`): The beta schedule, a mapping from a beta range to a sequence of betas for stepping the model. Choose from - `linear`, `scaled_linear`, or `squaredcos_cap_v2`. + `linear`, `scaled_linear`, `squaredcos_cap_v2`, or `sigmoid`. trained_betas (`np.ndarray`, *optional*): An array of betas to pass directly to the constructor without using `beta_start` and `beta_end`. variance_type (`str`, defaults to `"fixed_small"`):