Diffusers

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v0.40.0).

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.

CosineDPMSolverMultistepScheduler

class diffusers.CosineDPMSolverMultistepScheduler

< source >

( *args**kwargs )

SchedulerOutput

class diffusers.schedulers.scheduling_utils.SchedulerOutput

< source >

( 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.

Update on GitHub

←ConsistencyDecoderScheduler DDIMInverseScheduler→