Skip to content

Commit 4f3ec53

Browse files
authored
Add sigmoid scheduler in scheduling_ddpm.py docs (#10648)
Sigmoid scheduler in scheduling_ddpm.py docs
1 parent 07860f9 commit 4f3ec53

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/schedulers/scheduling_ddpm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin):
142142
The final `beta` value.
143143
beta_schedule (`str`, defaults to `"linear"`):
144144
The beta schedule, a mapping from a beta range to a sequence of betas for stepping the model. Choose from
145-
`linear`, `scaled_linear`, or `squaredcos_cap_v2`.
145+
`linear`, `scaled_linear`, `squaredcos_cap_v2`, or `sigmoid`.
146146
trained_betas (`np.ndarray`, *optional*):
147147
An array of betas to pass directly to the constructor without using `beta_start` and `beta_end`.
148148
variance_type (`str`, defaults to `"fixed_small"`):

0 commit comments

Comments
 (0)