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

CosineDPMSolverMultistepScheduler는 Nichol과 Dhariwal(2021)이 제안한 코사인 스케줄을 적용한 DPMSolverMultistepScheduler의 변형입니다. 이 스케줄러는 Stable Audio Open 논문과 Stability-AI/stable-audio-tool 코드베이스에서 사용됩니다.

이 스케줄러는 Yoach Lacombe님이 기여했습니다.

CosineDPMSolverMultistepScheduler

클래스 diffusers.CosineDPMSolverMultistepScheduler

< >

( *args**kwargs )

SchedulerOutput

class diffusers.schedulers.scheduling_utils.SchedulerOutput

< >

( prev_sample: Tensor )

매개변수

  • prev_sample (이미지의 경우 torch.Tensor, shape (batch_size, num_channels, height, width)) — 이전 timestep의 계산된 샘플 (x_{t-1}). prev_sample은 denoising 루프에서 다음 모델 입력으로 사용해야 합니다.

스케줄러의 step 함수 출력을 위한 베이스 클래스.

Update on GitHub