Struct candle_transformers::models::stable_diffusion::ddpm::DDPMScheduler

pub struct DDPMScheduler {
    pub config: DDPMSchedulerConfig,
    /* private fields */
}

Fields

config: DDPMSchedulerConfig

Implementations

impl DDPMScheduler

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

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 ) -> Result<Tensor>

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

pub fn init_noise_sigma(&self) -> f64