diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-09-18 16:51:38 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-18 16:51:38 +0100 |
| commit | 82a98f6da0e0d8273bcb6cf5766bc25a58843f80 (patch) | |
| tree | 306e6f0471a0293e0c7aa7dda6a2529ea94708da /candle-transformers/src/models/wuerstchen | |
| parent | 5082954c523c82a76ce91622ae9c3966dce0dc89 (diff) | |
| download | candle-82a98f6da0e0d8273bcb6cf5766bc25a58843f80.tar.gz candle-82a98f6da0e0d8273bcb6cf5766bc25a58843f80.tar.bz2 candle-82a98f6da0e0d8273bcb6cf5766bc25a58843f80.zip | |
Prior denoising. (#889)
Diffstat (limited to 'candle-transformers/src/models/wuerstchen')
| -rw-r--r-- | candle-transformers/src/models/wuerstchen/ddpm.rs | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/candle-transformers/src/models/wuerstchen/ddpm.rs b/candle-transformers/src/models/wuerstchen/ddpm.rs index f4f16bfb..80640072 100644 --- a/candle-transformers/src/models/wuerstchen/ddpm.rs +++ b/candle-transformers/src/models/wuerstchen/ddpm.rs @@ -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; |