wassname/denoising-diffusion-pytorch
1
0
Fork 0
mirror of https://github.com/wassname/denoising-diffusion-pytorch.git synced 2026-09-12 12:22:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix bug in elucidating sampling

Browse Source
This commit is contained in:
Phil Wang
2022-06-28 17:52:02 -07:00
parent 86fcb6785b
commit 8408775cfc
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
denoising_diffusion_pytorch/elucidated_diffusion.py
+1 -1
View File
@@ -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