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

move checkbox out of cached resource

Browse Source
This commit is contained in:
Stephan Auerhahn
2023-08-10 12:44:48 -07:00
parent a25662e969
commit b3866d1218
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/demo/sampling.py
View File

@@ -237,8 +237,12 @@ if __name__ == "__main__":
)
seed_everything(seed)
lowvram_mode = st.checkbox("Low vram mode", True)
save_locally, save_path = init_save_locally(os.path.join(SAVE_PATH, str(version)))
state = init_st(model_specs[version_enum], load_filter=True)
state = init_st(
model_specs[version_enum], load_filter=True, lowvram_mode=lowvram_mode
)
model = state["model"]
is_legacy = specs.is_legacy