pub struct EdmDpmMultistepScheduler { /* private fields */ }Implementations§
Source§impl EdmDpmMultistepScheduler
impl EdmDpmMultistepScheduler
pub fn new( config: UniPCSchedulerConfig, num_inference_steps: usize, ) -> Result<Self>
Trait Implementations§
Source§impl Scheduler for EdmDpmMultistepScheduler
impl Scheduler for EdmDpmMultistepScheduler
fn step( &mut self, model_output: &Tensor, timestep: usize, sample: &Tensor, ) -> Result<Tensor>
fn scale_model_input(&self, sample: Tensor, _timestep: usize) -> Result<Tensor>
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 EdmDpmMultistepScheduler
impl !RefUnwindSafe for EdmDpmMultistepScheduler
impl Send for EdmDpmMultistepScheduler
impl Sync for EdmDpmMultistepScheduler
impl Unpin for EdmDpmMultistepScheduler
impl !UnwindSafe for EdmDpmMultistepScheduler
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
§impl<T> Instrument for T
impl<T> Instrument for T
§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more