Buckets:

|
download
raw
1.13 kB

LMSDiscreteScheduler

LMSDiscreteScheduler is a linear multistep scheduler for discrete beta schedules. The scheduler is ported from and created by Katherine Crowson, and the original implementation can be found at crowsonkb/k-diffusion.

LMSDiscreteScheduler[[diffusers.LMSDiscreteScheduler]]

LMSDiscreteSchedulerOutput[[diffusers.schedulers.scheduling_lms_discrete.LMSDiscreteSchedulerOutput]]

  • prev_sample (torch.Tensor of shape (batch_size, num_channels, height, width) for images) -- Computed sample (x_{t-1}) of previous timestep. prev_sample should be used as next model input in the denoising loop.
  • pred_original_sample (torch.Tensor of shape (batch_size, num_channels, height, width) for images) -- The predicted denoised sample (x_{0}) based on the model output from the current timestep. pred_original_sample can be used to preview progress or for guidance.

Output class for the scheduler's step function output.

Xet Storage Details

Size:
1.13 kB
·
Xet hash:
7098c9439e015d9669011dea6fe19df10ceccac9d4ed0695fa88c164a1810348

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.