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