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

pass options into state2 init

Browse Source
This commit is contained in:
Stephan Auerhahn
2023-08-10 15:06:55 -07:00
parent cd81956241
commit d6f2b78994
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/demo/sampling.py
View File

@@ -279,7 +279,9 @@ if __name__ == "__main__":
st.write("**Refiner Options:**")
specs2 = model_specs[version2]
state2 = init_st(specs2, load_filter=False)
state2 = init_st(
specs2, load_filter=False, use_fp16=use_fp16, enable_swap=enable_swap
)
params2 = state2["params"]
params2.img2img_strength = st.number_input(