Fix incorrect spec_min and spec_max

openvpi · Apr 16, 2023 · 48e9474 · 48e9474
1 parent 20d0786
commit 48e9474
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/diffusion/ddpm.py b/modules/diffusion/ddpm.py
@@ -303,7 +303,7 @@ def __init__(self, vmin, vmax, num_bins, deviation, timesteps=1000, k_step=1000,
         super().__init__(
             num_bins, timesteps=timesteps, k_step=k_step,
             denoiser_type=denoiser_type, denoiser_args=denoiser_args,
-            betas=betas, spec_min=[0.], spec_max=[1.]
+            betas=betas, spec_min=[vmin], spec_max=[vmax]
         )
         self.vmin = vmin
         self.vmax = vmax