pub struct DDIMSchedulerConfig {
pub beta_start: f64,
pub beta_end: f64,
pub beta_schedule: BetaSchedule,
pub eta: f64,
pub steps_offset: usize,
pub prediction_type: PredictionType,
pub train_timesteps: usize,
pub timestep_spacing: TimestepSpacing,
}
Expand description
The configuration for the DDIM scheduler.
Fields§
§beta_start: f64
The value of beta at the beginning of training.
beta_end: f64
The value of beta at the end of training.
beta_schedule: BetaSchedule
How beta evolved during training.
eta: f64
The amount of noise to be added at each step.
steps_offset: usize
Adjust the indexes of the inference schedule by this value.
prediction_type: PredictionType
prediction type of the scheduler function, one of epsilon
(predicting
the noise of the diffusion process), sample
(directly predicting the noisy sample) or
v_prediction` (see section 2.4 https://imagen.research.google/video/paper.pdf)
train_timesteps: usize
number of diffusion steps used to train the model
timestep_spacing: TimestepSpacing
time step spacing for the diffusion process
Trait Implementations§
source§impl Clone for DDIMSchedulerConfig
impl Clone for DDIMSchedulerConfig
source§fn clone(&self) -> DDIMSchedulerConfig
fn clone(&self) -> DDIMSchedulerConfig
Returns a copy of the value. Read more
1.0.0 · source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source
. Read moresource§impl Debug for DDIMSchedulerConfig
impl Debug for DDIMSchedulerConfig
source§impl Default for DDIMSchedulerConfig
impl Default for DDIMSchedulerConfig
impl Copy for DDIMSchedulerConfig
Auto Trait Implementations§
impl Freeze for DDIMSchedulerConfig
impl RefUnwindSafe for DDIMSchedulerConfig
impl Send for DDIMSchedulerConfig
impl Sync for DDIMSchedulerConfig
impl Unpin for DDIMSchedulerConfig
impl UnwindSafe for DDIMSchedulerConfig
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more