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

remove extra init

Browse Source
This commit is contained in:
Stephan Auerhahn
2023-08-12 05:42:22 -07:00
parent 65c6ec1cec
commit e32972b85b
1 changed files with 5 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
scripts/demo/streamlit_helpers.py
View File

@@ -132,9 +132,7 @@ def show_samples(samples, outputs):
def get_guider(params: SamplingParams, key=1) -> SamplingParams:
params.guider = Guider(
st.sidebar.selectbox(
f"Discretization #{key}", [member.value for member in Guider]
)
st.sidebar.selectbox(f"Discretization #{key}", [member.value for member in Guider])
)
if params.guider == Guider.VANILLA:
@@ -161,18 +159,12 @@ def init_sampling(
key=1,
specify_num_samples=True,
) -> Tuple[SamplingParams, int, int]:
params = SamplingParams(img2img_strength=params.img2img_strength)
num_rows, num_cols = 1, 1
if specify_num_samples:
num_cols = st.number_input(
f"num cols #{key}", value=2, min_value=1, max_value=10
)
num_cols = st.number_input(f"num cols #{key}", value=2, min_value=1, max_value=10)
params.steps = int(
st.sidebar.number_input(
f"steps #{key}", value=params.steps, min_value=1, max_value=1000
)
st.sidebar.number_input(f"steps #{key}", value=params.steps, min_value=1, max_value=1000)
)
params.sampler = Sampler(
@@ -220,15 +212,11 @@ def get_sampler(params: SamplingParams, key=1) -> SamplingParams:
)
elif params.sampler in (Sampler.EULER_ANCESTRAL, Sampler.DPMPP2S_ANCESTRAL):
params.s_noise = st.sidebar.number_input(
"s_noise", value=params.s_noise, min_value=0.0
)
params.s_noise = st.sidebar.number_input("s_noise", value=params.s_noise, min_value=0.0)
params.eta = st.sidebar.number_input("eta", value=params.eta, min_value=0.0)
elif params.sampler == Sampler.LINEAR_MULTISTEP:
params.order = int(
st.sidebar.number_input("order", value=params.order, min_value=1)
)
params.order = int(st.sidebar.number_input("order", value=params.order, min_value=1))
return params