From 8408775cfc0c3830f01d2ea3bd83b6b48c45cb42 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Tue, 28 Jun 2022 17:52:02 -0700 Subject: [PATCH] fix bug in elucidating sampling --- denoising_diffusion_pytorch/elucidated_diffusion.py | 2 +- setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/denoising_diffusion_pytorch/elucidated_diffusion.py b/denoising_diffusion_pytorch/elucidated_diffusion.py index 740a1fe..8052df0 100644 --- a/denoising_diffusion_pytorch/elucidated_diffusion.py +++ b/denoising_diffusion_pytorch/elucidated_diffusion.py @@ -168,7 +168,7 @@ class ElucidatedDiffusion(nn.Module): for sigma, sigma_next, gamma in tqdm(sigmas_and_gammas, desc = 'sampling time step'): sigma, sigma_next, gamma = map(lambda t: t.item(), (sigma, sigma_next, gamma)) - eps = gamma * torch.randn(shape, device = self.device) + eps = self.S_noise * torch.randn(shape, device = self.device) # stochastic sampling sigma_hat = sigma + gamma * sigma images_hat = images + sqrt(sigma_hat ** 2 - sigma ** 2) * eps diff --git a/setup.py b/setup.py index ff1eb12..b4db8c0 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.23.0', + version = '0.23.1', license='MIT', description = 'Denoising Diffusion Probabilistic Models - Pytorch', author = 'Phil Wang',