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

The CosineDPMSolverMultistepScheduler is a variant of DPMSolverMultistepScheduler with cosine schedule, proposed by Nichol and Dhariwal (2021). It is being used in the Stable Audio Open paper and the Stability-AI/stable-audio-tool codebase.

This scheduler was contributed by Yoach Lacombe.

( *args**kwargs )

( prev_sample: Tensor )

Parameters

prev_sample (torch.Tensor of shape (batch_size, num_channels, height, width) for images) — Computed sample (x_{t-1}) of previous timestep. prev_sample should be used as next model input in the denoising loop.

Base class for the output of a scheduler’s step function.