Diffusers documentation

CosineDPMSolverMultistepScheduler

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

CosineDPMSolverMultistepScheduler

CosineDPMSolverMultistepScheduler — это вариант DPMSolverMultistepScheduler с косинусной раскладкой, предложенный Nichol и Dhariwal (2021). Он используется в статье Stable Audio Open и в кодовой базе Stability-AI/stable-audio-tool.

Этот планировщик был предложен Yoach Lacombe.

CosineDPMSolverMultistepScheduler

класс diffusers.CosineDPMSolverMultistepScheduler

< >

( *args**kwargs )

SchedulerOutput

класс diffusers.schedulers.scheduling_utils.SchedulerOutput

< >

( prev_sample: Tensor )

Параметры

  • prev_sample (torch.Tensor формы (batch_size, num_channels, height, width) для изображений) — Вычисленная выборка (x_{t-1}) предыдущего шага. prev_sample следует использовать как следующий вход модели в цикле шумоподавления.

Базовый класс для результата функции step планировщика.

Update on GitHub