AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Add sigmoid scheduler in scheduling_ddpm.py docs (#10648)

Browse Source 
Sigmoid scheduler in scheduling_ddpm.py docs
This commit is contained in:
Jacob Helwig
2025-01-26 17:37:20 -06:00
committed by GitHub
parent 07860f9916
commit 4f3ec5364e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/schedulers/scheduling_ddpm.py
View File

@@ -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"`):