Diffusers documentation

LMSDiscreteScheduler

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

LMSDiscreteScheduler

LMSDiscreteScheduler es un programador de múltiples pasos lineales para programaciones beta discretas. El programador fue portado y creado por Katherine Crowson, y la implementación original se puede encontrar en crowsonkb/k-diffusion.

LMSDiscreteScheduler

clase diffusers.LMSDiscreteScheduler

< >

( *args**kwargs )

LMSDiscreteSchedulerOutput

clase diffusers.schedulers.scheduling_lms_discrete.LMSDiscreteSchedulerOutput

< >

( prev_sample: Tensorpred_original_sample: typing.Optional[torch.Tensor] = None )

Parámetros

  • prev_sample (torch.Tensor de forma (batch_size, num_channels, height, width) para imágenes) — Muestra calculada (x_{t-1}) del paso de tiempo anterior. prev_sample debe usarse como entrada del modelo en el bucle de desruido.
  • pred_original_sample (torch.Tensor de forma (batch_size, num_channels, height, width) para imágenes) — La muestra desruidada predicha (x_{0}) basada en la salida del modelo del paso de tiempo actual. pred_original_sample se puede usar para previsualizar el progreso o para la guía.

Clase de salida para la función step del programador.

Update on GitHub