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

Le CosineDPMSolverMultistepScheduler est une variante du DPMSolverMultistepScheduler avec un planning cosinus, proposée par Nichol et Dhariwal (2021). Il est utilisé dans l'article Stable Audio Open et dans la base de code Stability-AI/stable-audio-tool.

Ce planificateur a été contribué par Yoach Lacombe.

CosineDPMSolverMultistepScheduler

classe diffusers.CosineDPMSolverMultistepScheduler

< >

( *args**kwargs )

SchedulerOutput

classe diffusers.schedulers.scheduling_utils.SchedulerOutput

< >

( prev_sample: Tensor )

Paramètres

  • prev_sample (torch.Tensor de forme (batch_size, num_channels, height, width) pour les images) — Échantillon calculé (x_{t-1}) de l'étape temporelle précédente. prev_sample doit être utilisé comme entrée du modèle suivante dans la boucle de débruitage.

Classe de base pour la sortie de la fonction step d’un planificateur.

Update on GitHub