aljaz/generative-models
1
0
Fork 0
mirror of https://github.com/Stability-AI/generative-models.git synced 2025-12-21 07:14:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix EDMDiscretization sigma_min for correct sampling noise scheduling (#114)

Browse Source
This commit is contained in:
Vitaly Bondar
2023-08-17 18:48:30 +03:00
committed by GitHub
parent 45c443b316
commit 477d8b9a77
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sgm/modules/diffusionmodules/discretizer.py
View File

@@ -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