diffusers/pr_14421/en/api/schedulers/cosine_dpm.md

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[[diffusers.CosineDPMSolverMultistepScheduler]]

diffusers.CosineDPMSolverMultistepScheduler[[diffusers.CosineDPMSolverMultistepScheduler]]

diffusers.CosineDPMSolverMultistepScheduler(*args, **kwargs)

Source

SchedulerOutput[[diffusers.schedulers.scheduling_utils.SchedulerOutput]]

diffusers.schedulers.scheduling_utils.SchedulerOutput[[diffusers.schedulers.scheduling_utils.SchedulerOutput]]

diffusers.schedulers.scheduling_utils.SchedulerOutput(prev_sample: Tensor)

Source

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.