diff options
Diffstat (limited to 'candle-transformers/src/models/stable_diffusion/schedulers.rs')
| -rw-r--r-- | candle-transformers/src/models/stable_diffusion/schedulers.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-transformers/src/models/stable_diffusion/schedulers.rs b/candle-transformers/src/models/stable_diffusion/schedulers.rs index 1d39037f..1ce94ca2 100644 --- a/candle-transformers/src/models/stable_diffusion/schedulers.rs +++ b/candle-transformers/src/models/stable_diffusion/schedulers.rs @@ -19,7 +19,7 @@ pub trait Scheduler { fn scale_model_input(&self, sample: Tensor, _timestep: usize) -> Result<Tensor>; - fn step(&self, model_output: &Tensor, timestep: usize, sample: &Tensor) -> Result<Tensor>; + fn step(&mut self, model_output: &Tensor, timestep: usize, sample: &Tensor) -> Result<Tensor>; } /// This represents how beta ranges from its minimum value to the maximum |