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

O CosineDPMSolverMultistepScheduler é uma variante do DPMSolverMultistepScheduler com agendamento coseno, proposta por Nichol e Dhariwal (2021). Ele é utilizado no paper Stable Audio Open e na base de código Stability-AI/stable-audio-tool.

Este scheduler foi contribuído por Yoach Lacombe.

( *args**kwargs )

( prev_sample: Tensor )

Parâmetros

prev_sample (torch.Tensor de forma (batch_size, num_channels, height, width) para imagens) — Amostra calculada (x_{t-1}) do passo de tempo anterior. prev_sample deve ser usado como entrada do modelo no loop de remoção de ruído.

Classe base para a saída da função step de um agendador.