Struct DDPMWScheduler 

pub struct DDPMWScheduler {
    pub config: DDPMWSchedulerConfig,
    /* private fields */
}

Fields

config: DDPMWSchedulerConfig

Implementations

impl DDPMWScheduler

pub fn new(inference_steps: usize, config: DDPMWSchedulerConfig) -> Result<Self>

pub fn timesteps(&self) -> &[f64]

pub fn scale_model_input(&self, sample: Tensor, _timestep: usize) -> Tensor

Ensures interchangeability with schedulers that need to scale the denoising model input depending on the current timestep.

pub fn step( &self, model_output: &Tensor, ts: f64, sample: &Tensor, ) -> Result<Tensor>

pub fn init_noise_sigma(&self) -> f64