reminder to future self

2022-07-20 10:19:15 -06:00 · 2022-07-20 10:19:15 -06:00 · 9a37f3ba42
commit 9a37f3ba42
parent 15decfdb98
1 changed files with 1 additions and 0 deletions
--- a/codes/models/diffusion/gaussian_diffusion.py
+++ b/codes/models/diffusion/gaussian_diffusion.py
@ -350,6 +350,7 @@ class GaussianDiffusion:
            else:
                cfk = self.conditioning_free_k
            model_output = (1 + cfk) * model_output - cfk * model_output_no_conditioning
+            # TODO: combine variance predictions here similarly.

        def process_xstart(x):
            if denoised_fn is not None: