From 9a37f3ba42b5d4f51dd9d48df2bf41d20d193ead Mon Sep 17 00:00:00 2001 From: James Betker Date: Wed, 20 Jul 2022 10:19:15 -0600 Subject: [PATCH] reminder to future self --- codes/models/diffusion/gaussian_diffusion.py | 1 + 1 file changed, 1 insertion(+) diff --git a/codes/models/diffusion/gaussian_diffusion.py b/codes/models/diffusion/gaussian_diffusion.py index 42ceb144..c20b7b30 100644 --- 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: