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

Changed LegacyDDPMDiscretization for sampling

Browse Source
This commit is contained in:
Tim
2023-06-29 15:47:37 -07:00
parent 613af104c6
commit e9869d7822
2 changed files with 24 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/demo/streamlit_helpers.py
View File

@@ -325,10 +325,8 @@ def init_sampling(
def get_discretization(discretization, key=1):
if discretization == "LegacyDDPMDiscretization":
use_new_range = st.checkbox(f"Start from highest noise level? #{key}", False)
discretization_config = {
"target": "sgm.modules.diffusionmodules.discretizer.LegacyDDPMDiscretization",
"params": {"legacy_range": not use_new_range},
}
elif discretization == "EDMDiscretization":
sigma_min = st.number_input(f"sigma_min #{key}", value=0.03) # 0.0292