Struct DDIMScheduler

pub struct DDIMScheduler {
    pub config: DDIMSchedulerConfig,
    /* private fields */
}

The DDIM scheduler.

Fields

config: DDIMSchedulerConfig

Implementations

impl DDIMScheduler

pub fn new(inference_steps: usize, config: DDIMSchedulerConfig) -> Self

Creates a new DDIM scheduler given the number of steps to be used for inference as well as the number of steps that was used during training.

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

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, timestep: usize, sample: &Tensor, ) -> Tensor

Performs a backward step during inference.

pub fn add_noise( &self, original: &Tensor, noise: Tensor, timestep: usize, ) -> Tensor

pub fn init_noise_sigma(&self) -> f64

Trait Implementations

impl Clone for DDIMScheduler

fn clone(&self) -> DDIMScheduler

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for DDIMScheduler

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.