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

low vram checkbox fix, remove magic strings

Browse Source
This commit is contained in:
Stephan Auerhahn
2023-08-10 12:40:32 -07:00
parent 26b10f56f3
commit a25662e969
2 changed files with 14 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
scripts/demo/sampling.py
View File

@@ -23,7 +23,6 @@ from scripts.demo.streamlit_helpers import (
init_sampling,
init_save_locally,
init_st,
set_lowvram_mode,
show_samples,
)
@@ -205,6 +204,16 @@ def apply_refiner(
return samples
sdxl_base_model_list = [
ModelArchitecture.SDXL_V1_BASE,
ModelArchitecture.SDXL_V0_9_BASE,
]
sdxl_refiner_model_list = [
ModelArchitecture.SDXL_V1_REFINER,
ModelArchitecture.SDXL_V0_9_REFINER,
]
if __name__ == "__main__":
st.title("Stable Diffusion")
version = st.selectbox(
@@ -217,9 +226,7 @@ if __name__ == "__main__":
mode = st.radio("Mode", ("txt2img", "img2img"), 0)
st.write("__________________________")
set_lowvram_mode(st.checkbox("Low vram mode", True))
if str(version).startswith("stable-diffusion-xl"):
if version_enum in sdxl_base_model_list:
add_pipeline = st.checkbox("Load SDXL-refiner?", False)
st.write("__________________________")
else:
@@ -253,10 +260,7 @@ if __name__ == "__main__":
version2 = ModelArchitecture(
st.selectbox(
"Refiner:",
[
ModelArchitecture.SDXL_V1_REFINER.value,
ModelArchitecture.SDXL_V0_9_REFINER.value,
],
[member.value for member in sdxl_refiner_model_list],
)
)
st.warning(