diff options
Diffstat (limited to 'candle-transformers/src/models/stable_diffusion/ddpm.rs')
| -rw-r--r-- | candle-transformers/src/models/stable_diffusion/ddpm.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-transformers/src/models/stable_diffusion/ddpm.rs b/candle-transformers/src/models/stable_diffusion/ddpm.rs index d393f39a..42a0dc7e 100644 --- a/candle-transformers/src/models/stable_diffusion/ddpm.rs +++ b/candle-transformers/src/models/stable_diffusion/ddpm.rs @@ -104,7 +104,7 @@ impl DDPMScheduler { }; let current_beta_t = 1. - alpha_prod_t / alpha_prod_t_prev; - // For t > 0, compute predicted variance βt (see formula (6) and (7) from https://arxiv.org/pdf/2006.11239.pdf) + // For t > 0, compute predicted variance βt (see formula (6) and (7) from [the pdf](https://arxiv.org/pdf/2006.11239.pdf)) // and sample from it to get previous sample // x_{t-1} ~ N(pred_prev_sample, variance) == add variance to pred_sample let variance = (1. - alpha_prod_t_prev) / (1. - alpha_prod_t) * current_beta_t; |