fix EDMDiscretization sigma_min for correct sampling noise scheduling (…

…Stability-AI#114)
SevanBrodjian · Aug 17, 2023 · c4b74b9 · c4b74b9
1 parent 7dcf5be
commit c4b74b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sgm/modules/diffusionmodules/discretizer.py b/sgm/modules/diffusionmodules/discretizer.py
@@ -26,7 +26,7 @@ def get_sigmas(self, n, device):
 
 
 class EDMDiscretization(Discretization):
-    def __init__(self, sigma_min=0.02, sigma_max=80.0, rho=7.0):
+    def __init__(self, sigma_min=0.002, sigma_max=80.0, rho=7.0):
         self.sigma_min = sigma_min
         self.sigma_max = sigma_max
         self.rho = rho