Skip to content

Commit 477d8b9

Browse files
authored
fix EDMDiscretization sigma_min for correct sampling noise scheduling (#114)
1 parent 45c443b commit 477d8b9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sgm/modules/diffusionmodules/discretizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def get_sigmas(self, n, device):
2626

2727

2828
class EDMDiscretization(Discretization):
29-
def __init__(self, sigma_min=0.02, sigma_max=80.0, rho=7.0):
29+
def __init__(self, sigma_min=0.002, sigma_max=80.0, rho=7.0):
3030
self.sigma_min = sigma_min
3131
self.sigma_max = sigma_max
3232
self.rho = rho

0 commit comments

Comments
 (0)