Unverified Commit 03373de0 authored by Steven Liu's avatar Steven Liu Committed by GitHub
Browse files

[docs] Add missing parameter (#6775)

add missing param
parent 56bea6b4
...@@ -143,6 +143,8 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin): ...@@ -143,6 +143,8 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin):
beta_schedule (`str`, defaults to `"linear"`): 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 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`, or `squaredcos_cap_v2`.
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"`): variance_type (`str`, defaults to `"fixed_small"`):
Clip the variance when adding noise to the denoised sample. Choose from `fixed_small`, `fixed_small_log`, Clip the variance when adding noise to the denoised sample. Choose from `fixed_small`, `fixed_small_log`,
`fixed_large`, `fixed_large_log`, `learned` or `learned_range`. `fixed_large`, `fixed_large_log`, `learned` or `learned_range`.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment