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

fix off by one

Browse Source
This commit is contained in:
Phil Wang committed 2022-06-28 16:05:21 -07:00
1 parent a8403b83fe
commit b87ea27781
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
denoising_diffusion_pytorch/elucidated_diffusion.py
+1 -1
View File
@@ -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