Update variance_schedule.py

mathpluscode · Nov 30, 2023 · 16f1e20 · 16f1e20
1 parent 0e3bd82
commit 16f1e20
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/imgx/diffusion/gaussian/variance_schedule.py b/imgx/diffusion/gaussian/variance_schedule.py
@@ -107,6 +107,8 @@ def downsample_beta_schedule(
 
     After down-sampling, the first and last values of alphas_cumprod are kept.
 
+    https://arxiv.org/abs/2102.09672
+
     Args:
         betas: beta schedule, shape (num_timesteps,).
             Values are in ascending order.