Skip to content

Commit 03373de

Browse files
authored
[docs] Add missing parameter (#6775)
add missing param
1 parent 56bea6b commit 03373de

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/diffusers/schedulers/scheduling_ddpm.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -143,6 +143,8 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin):
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
145145
`linear`, `scaled_linear`, or `squaredcos_cap_v2`.
146+
trained_betas (`np.ndarray`, *optional*):
147+
An array of betas to pass directly to the constructor without using `beta_start` and `beta_end`.
146148
variance_type (`str`, defaults to `"fixed_small"`):
147149
Clip the variance when adding noise to the denoised sample. Choose from `fixed_small`, `fixed_small_log`,
148150
`fixed_large`, `fixed_large_log`, `learned` or `learned_range`.

0 commit comments

Comments
 (0)