From 477d8b9a7730d9b2e92b326a770c0420d00308c9 Mon Sep 17 00:00:00 2001 From: Vitaly Bondar Date: Thu, 17 Aug 2023 18:48:30 +0300 Subject: [PATCH] fix EDMDiscretization sigma_min for correct sampling noise scheduling (#114) --- sgm/modules/diffusionmodules/discretizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sgm/modules/diffusionmodules/discretizer.py b/sgm/modules/diffusionmodules/discretizer.py index 4135ac9..02add60 100644 --- a/sgm/modules/diffusionmodules/discretizer.py +++ b/sgm/modules/diffusionmodules/discretizer.py @@ -26,7 +26,7 @@ class Discretization: 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