Struct candle_transformers::models::stable_diffusion::ddim::DDIMScheduler
source · pub struct DDIMScheduler {
pub config: DDIMSchedulerConfig,
/* private fields */
}
Expand description
The DDIM scheduler.
Fields§
§config: DDIMSchedulerConfig
Trait Implementations§
source§impl Clone for DDIMScheduler
impl Clone for DDIMScheduler
source§fn clone(&self) -> DDIMScheduler
fn clone(&self) -> DDIMScheduler
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 DDIMScheduler
impl Debug for DDIMScheduler
source§impl Scheduler for DDIMScheduler
impl Scheduler for DDIMScheduler
source§fn step(
&self,
model_output: &Tensor,
timestep: usize,
sample: &Tensor
) -> Result<Tensor>
fn step( &self, model_output: &Tensor, timestep: usize, sample: &Tensor ) -> Result<Tensor>
Performs a backward step during inference.
source§fn scale_model_input(&self, sample: Tensor, _timestep: usize) -> Result<Tensor>
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
Auto Trait Implementations§
impl Freeze for DDIMScheduler
impl RefUnwindSafe for DDIMScheduler
impl Send for DDIMScheduler
impl Sync for DDIMScheduler
impl Unpin for DDIMScheduler
impl UnwindSafe for DDIMScheduler
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