From b3866d121890034ff26a1c815223af5847fdd7fd Mon Sep 17 00:00:00 2001 From: Stephan Auerhahn Date: Thu, 10 Aug 2023 12:44:48 -0700 Subject: [PATCH] move checkbox out of cached resource --- scripts/demo/sampling.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/scripts/demo/sampling.py b/scripts/demo/sampling.py index 4dca18d..9bf0dff 100644 --- a/scripts/demo/sampling.py +++ b/scripts/demo/sampling.py @@ -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