diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-08-06 22:39:53 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-06 21:39:53 +0100 |
| commit | 141df4ad2b80f0691450f32b28bcc2314301050b (patch) | |
| tree | 357a62dd4bce15a2f43dc86d507f62748d61a820 /candle-examples/examples/stable-diffusion/ddim.rs | |
| parent | 166bfd5847144abec227836e497b509625470535 (diff) | |
| download | candle-141df4ad2b80f0691450f32b28bcc2314301050b.tar.gz candle-141df4ad2b80f0691450f32b28bcc2314301050b.tar.bz2 candle-141df4ad2b80f0691450f32b28bcc2314301050b.zip | |
Main diffusion loop for the SD example. (#332)
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. |