From e147839d74f3af949e0048e64c0a33402839d6f9 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Thu, 12 May 2022 10:08:53 -0700 Subject: [PATCH] make sure to clip when sampling from gaussian diffusion with learned variance --- denoising_diffusion_pytorch/learned_gaussian_diffusion.py | 4 ++++ setup.py | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/denoising_diffusion_pytorch/learned_gaussian_diffusion.py b/denoising_diffusion_pytorch/learned_gaussian_diffusion.py index 94708fc..45c040e 100644 --- a/denoising_diffusion_pytorch/learned_gaussian_diffusion.py +++ b/denoising_diffusion_pytorch/learned_gaussian_diffusion.py @@ -107,6 +107,10 @@ class LearnedGaussianDiffusion(GaussianDiffusion): model_variance = model_log_variance.exp() x_start = self.predict_start_from_noise(x, t, pred_noise) + + if clip_denoised: + x_start.clamp_(-1., 1.) + model_mean, _, _ = self.q_posterior(x_start, x, t) return model_mean, model_variance, model_log_variance diff --git a/setup.py b/setup.py index f92a74d..2b33658 100644 --- a/setup.py +++ b/setup.py @@ -3,7 +3,7 @@ from setuptools import setup, find_packages setup( name = 'denoising-diffusion-pytorch', packages = find_packages(), - version = '0.14.0', + version = '0.14.1', license='MIT', description = 'Denoising Diffusion Probabilistic Models - Pytorch', author = 'Phil Wang',