Skip to content

Commit a1e7363

Browse files
committed
Update documentation.
1 parent 6376fec commit a1e7363

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/schedulers/scheduling_ddpm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def betas_for_alpha_bar(
6363
max_beta (`float`): the maximum beta to use; use values lower than 1 to
6464
prevent singularities.
6565
alpha_transform_type (`str`, *optional*, default to `cosine`): the type of noise schedule for alpha_bar.
66-
Choose from `cosine` or `exp`
66+
Choose from `cosine`, `laplace` or `exp`
6767
6868
Returns:
6969
betas (`np.ndarray`): the betas used by the scheduler to step the model outputs
@@ -149,7 +149,7 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin):
149149
The final `beta` value.
150150
beta_schedule (`str`, defaults to `"linear"`):
151151
The beta schedule, a mapping from a beta range to a sequence of betas for stepping the model. Choose from
152-
`linear`, `scaled_linear`, `squaredcos_cap_v2`, or `sigmoid`.
152+
`linear`, `scaled_linear`, `squaredcos_cap_v2`, `laplace` or `sigmoid`.
153153
trained_betas (`np.ndarray`, *optional*):
154154
An array of betas to pass directly to the constructor without using `beta_start` and `beta_end`.
155155
variance_type (`str`, defaults to `"fixed_small"`):

0 commit comments

Comments
 (0)