From 4f3ec5364e1543ce0e0b866eeed239f1aedcb9f4 Mon Sep 17 00:00:00 2001 From: Jacob Helwig <60412857+JacobHelwig@users.noreply.github.com> Date: Sun, 26 Jan 2025 17:37:20 -0600 Subject: [PATCH] Add sigmoid scheduler in `scheduling_ddpm.py` docs (#10648) 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 eb40d79b9f..624d5a5cd4 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"`):