diff options
Diffstat (limited to 'candle-examples/examples/stable-diffusion/ddim.rs')
| -rw-r--r-- | candle-examples/examples/stable-diffusion/ddim.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-examples/examples/stable-diffusion/ddim.rs b/candle-examples/examples/stable-diffusion/ddim.rs index 9afff5aa..6eb6df44 100644 --- a/candle-examples/examples/stable-diffusion/ddim.rs +++ b/candle-examples/examples/stable-diffusion/ddim.rs @@ -100,8 +100,8 @@ impl DDIMScheduler { /// Ensures interchangeability with schedulers that need to scale the denoising model input /// depending on the current timestep. - pub fn scale_model_input(&self, sample: Tensor, _timestep: usize) -> Tensor { - sample + pub fn scale_model_input(&self, sample: Tensor, _timestep: usize) -> Result<Tensor> { + Ok(sample) } /// Performs a backward step during inference. |