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 là một trình lập lịch đa bước tuyến tính cho các lịch beta rời rạc. Trình lập lịch này được chuyển đổi và tạo bởi Katherine Crowson, và cài đặt gốc có thể được tìm thấy tại crowsonkb/k-diffusion.

LMSDiscreteScheduler

lớp diffusers.LMSDiscreteScheduler

< >

( *args**kwargs )

LMSDiscreteSchedulerOutput

lớp diffusers.schedulers.scheduling_lms_discrete.LMSDiscreteSchedulerOutput

< >

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

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.
  • pred_original_sample (torch.Tensor có hình dạng (batch_size, num_channels, height, width) đối với ảnh) — Mẫu khử nhiễu dự đoán (x_{0}) dựa trên đầu ra của mô hình tại bước thời gian hiện tại. pred_original_sample có thể được sử dụng để xem trước tiến độ hoặc cho việc dẫn hướng.

Lớp đầu ra cho kết quả của hàm step của trình lập lịch.

Update on GitHub