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

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.

( *args**kwargs )

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