From b87ea2778111f40ba74fa2b7508c0d70a7628d2f Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Tue, 28 Jun 2022 16:05:21 -0700 Subject: [PATCH] fix off by one --- denoising_diffusion_pytorch/elucidated_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/denoising_diffusion_pytorch/elucidated_diffusion.py b/denoising_diffusion_pytorch/elucidated_diffusion.py index 782121c..3204a97 100644 --- a/denoising_diffusion_pytorch/elucidated_diffusion.py +++ b/denoising_diffusion_pytorch/elucidated_diffusion.py @@ -116,7 +116,7 @@ class ElucidatedDiffusion(nn.Module): N = num_sample_steps inv_rho = 1 / rho - for i in range(num_sample_steps - 1): + for i in range(num_sample_steps): sigma_i = (sigma_max ** inv_rho + i / (N - 1) * (sigma_min ** inv_rho - sigma_max ** inv_rho)) ** rho gamma_i = gamma if S_tmin <= sigma_i <= S_tmax else 0. yield sigma_i, gamma_i