Diffusers documentation

DPMSolverSDEScheduler

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

DPMSolverSDEScheduler

O DPMSolverSDEScheduler é inspirado no amostrador estocástico do artigo Elucidating the Design Space of Diffusion-Based Generative Models, e o agendador foi portado e criado por Katherine Crowson.

DPMSolverSDEScheduler

classe diffusers.DPMSolverSDEScheduler

< >

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