huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-20 12:06:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Prior denoising. (#889)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-18 16:51:38 +01:00
committed by GitHub
parent 5082954c52
commit 82a98f6da0
2 changed files with 25 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-transformers/src/models/wuerstchen/ddpm.rs
View File

@ -38,6 +38,10 @@ impl DDPMWScheduler {
})
}
pub fn timesteps(&self) -> &[f64] {
&self.timesteps
}
fn alpha_cumprod(&self, t: f64) -> f64 {
let scaler = self.config.scaler;
let s = self.config.s;