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 là một biến thể của DPMSolverMultistepScheduler với lịch cosine, được đề xuất bởi Nichol và Dhariwal (2021). Nó được sử dụng trong bài báo Stable Audio Open và mã nguồn Stability-AI/stable-audio-tool.

Trình lập lịch này được đóng góp bởi Yoach Lacombe.

CosineDPMSolverMultistepScheduler

lớp diffusers.CosineDPMSolverMultistepScheduler

< >

( *args**kwargs )

SchedulerOutput

lớp diffusers.schedulers.scheduling_utils.SchedulerOutput

< >

( prev_sample: Tensor )

Tham số

  • prev_sample (torch.Tensor có shape (batch_size, num_channels, height, width) đối với ảnh) — Mẫu đã tính (x_{t-1}) của bước thời gian trước đó. prev_sample nên được sử dụng làm đầu vào mô hình tiếp theo trong vòng lặp khử nhiễu.

Lớp cơ sở cho kết quả của hàm step của một trình lập lịch.

Update on GitHub