Struct candle_transformers::models::stable_diffusion::ddpm::DDPMScheduler
source · pub struct DDPMScheduler {
pub config: DDPMSchedulerConfig,
/* private fields */
}
Fields§
§config: DDPMSchedulerConfig
Implementations§
source§impl DDPMScheduler
impl DDPMScheduler
pub fn new(inference_steps: usize, config: DDPMSchedulerConfig) -> Result<Self>
pub fn timesteps(&self) -> &[usize]
sourcepub fn scale_model_input(&self, sample: Tensor, _timestep: usize) -> Tensor
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
Auto Trait Implementations§
impl Freeze for DDPMScheduler
impl RefUnwindSafe for DDPMScheduler
impl Send for DDPMScheduler
impl Sync for DDPMScheduler
impl Unpin for DDPMScheduler
impl UnwindSafe for DDPMScheduler
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