Diffusers

Join the Hugging Face community

and get access to the augmented documentation experience

Collaborate on models, datasets and Spaces

Faster examples with accelerated inference

Switch between documentation themes

to get started

CosineDPMSolverMultistepScheduler

CosineDPMSolverMultistepScheduler adalah varian dari DPMSolverMultistepScheduler dengan jadwal kosinus, yang diusulkan oleh Nichol dan Dhariwal (2021). Hal ini digunakan dalam makalah Stable Audio Open dan basis kode Stability-AI/stable-audio-tool.

Scheduler ini disumbangkan oleh Yoach Lacombe.

( *args**kwargs )

( prev_sample: Tensor )

Parameter

prev_sample (torch.Tensor dengan bentuk (batch_size, num_channels, height, width) untuk gambar) — Contoh yang dihitung (x_{t-1}) dari langkah waktu sebelumnya. prev_sample harus digunakan sebagai input model berikutnya dalam loop denoising.

Kelas dasar untuk keluaran dari fungsi step sebuah scheduler.