Struct candle_transformers::models::stable_diffusion::ddim::DDIMScheduler

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

The DDIM scheduler.

Fields

config: DDIMSchedulerConfig

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.

impl Scheduler for DDIMScheduler

fn step( &self, model_output: &Tensor, timestep: usize, sample: &Tensor ) -> Result<Tensor>

Performs a backward step during inference.

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

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

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

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

fn init_noise_sigma(&self) -> f64