aljaz/generative-models
1
0
Fork 0
mirror of https://github.com/Stability-AI/generative-models.git synced 2026-02-06 14:24:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

context manager

Browse Source
This commit is contained in:
Stephan Auerhahn
2023-08-09 12:38:44 -07:00
parent a726ce3eb7
commit f86ffac274
2 changed files with 31 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/demo/streamlit_helpers.py
View File

@@ -200,12 +200,8 @@ def init_sampling(
def get_discretization(params: SamplingParams, key=1) -> SamplingParams:
if params.discretization == Discretization.EDM:
params.sigma_min = st.number_input(
f"sigma_min #{key}", value=params.sigma_min
) # 0.0292
params.sigma_max = st.number_input(
f"sigma_max #{key}", value=params.sigma_max
) # 14.6146
params.sigma_min = st.number_input(f"sigma_min #{key}", value=params.sigma_min)
params.sigma_max = st.number_input(f"sigma_max #{key}", value=params.sigma_max)
params.rho = st.number_input(f"rho #{key}", value=params.rho)
return params