Diffusers

LMSDiscreteScheduler

LMSDiscreteScheduler adalah penjadwal multistep linear untuk jadwal beta diskrit. Penjadwal ini dipindahkan dan dibuat oleh Katherine Crowson, dan implementasi aslinya dapat ditemukan di crowsonkb/k-diffusion.

LMSDiscreteScheduler

kelas diffusers.LMSDiscreteScheduler

< source >

( *args**kwargs )

LMSDiscreteSchedulerOutput

kelas diffusers.schedulers.scheduling_lms_discrete.LMSDiscreteSchedulerOutput

< source >

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

Parameter

prev_sample (torch.Tensor dengan bentuk (batch_size, num_channels, height, width) untuk gambar) — Contoh yang dihitung (x_{t-1}) dari langkah waktu sebelumnya. prev_sample harus digunakan sebagai input model berikutnya dalam loop denoising.
pred_original_sample (torch.Tensor dengan bentuk (batch_size, num_channels, height, width) untuk gambar) — Sampel denoised yang diprediksi (x_{0}) berdasarkan output model dari timestep saat ini. pred_original_sample dapat digunakan untuk pratinjau progres atau untuk guidance.

Kelas output untuk output fungsi step penjadwal.

Update on GitHub