Diffusers documentation

CosineDPMSolverMultistepScheduler

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v0.42.1).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

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.

CosineDPMSolverMultistepScheduler

classe diffusers.CosineDPMSolverMultistepScheduler

< >

( *args**kwargs )

SchedulerOutput

classe diffusers.schedulers.scheduling_utils.SchedulerOutput

< >

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

Update on GitHub