Skip to content

Commit

Permalink
Add sigmoid scheduler in scheduling_ddpm.py docs (huggingface#10648)
Browse files Browse the repository at this point in the history
Sigmoid scheduler in scheduling_ddpm.py docs
  • Loading branch information
JacobHelwig authored Jan 26, 2025
1 parent 07860f9 commit 4f3ec53
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/diffusers/schedulers/scheduling_ddpm.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"`):
Expand Down

0 comments on commit 4f3ec53

Please sign in to comment.