huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Main diffusion loop for the SD example. (#332)

Browse Source
This commit is contained in:
Laurent Mazare
2023-08-06 22:39:53 +02:00
committed by GitHub
parent 166bfd5847
commit 141df4ad2b
3 changed files with 242 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-examples/examples/stable-diffusion/ddim.rs
View File

@ -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.