Skip to content

Commit 7f6373d

Browse files
[Docs] Add sigmoid beta_scheduler to docstrings of relevant Schedulers (#3399)
* Add `sigmoid` beta scheduler to `DDPMScheduler` docstring * Add `sigmoid` beta scheduler to `RePaintScheduler` docstring --------- Co-authored-by: Patrick von Platen <[email protected]>
1 parent 3a237f4 commit 7f6373d

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/diffusers/schedulers/scheduling_ddpm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin):
9191
beta_end (`float`): the final `beta` value.
9292
beta_schedule (`str`):
9393
the beta schedule, a mapping from a beta range to a sequence of betas for stepping the model. Choose from
94-
`linear`, `scaled_linear`, or `squaredcos_cap_v2`.
94+
`linear`, `scaled_linear`, `squaredcos_cap_v2` or `sigmoid`.
9595
trained_betas (`np.ndarray`, optional):
9696
option to pass an array of betas directly to the constructor to bypass `beta_start`, `beta_end` etc.
9797
variance_type (`str`):

src/diffusers/schedulers/scheduling_repaint.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ class RePaintScheduler(SchedulerMixin, ConfigMixin):
8989
beta_end (`float`): the final `beta` value.
9090
beta_schedule (`str`):
9191
the beta schedule, a mapping from a beta range to a sequence of betas for stepping the model. Choose from
92-
`linear`, `scaled_linear`, or `squaredcos_cap_v2`.
92+
`linear`, `scaled_linear`, `squaredcos_cap_v2` or `sigmoid`.
9393
eta (`float`):
9494
The weight of noise for added noise in a diffusion step. Its value is between 0.0 and 1.0 -0.0 is DDIM and
9595
1.0 is DDPM scheduler respectively.

0 commit comments

Comments
 (0)