move checkbox out of cached resource

Stability-AI · Aug 10, 2023 · b3866d1 · b3866d1
1 parent a25662e
commit b3866d1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/scripts/demo/sampling.py b/scripts/demo/sampling.py
@@ -237,8 +237,12 @@ def apply_refiner(
     )
     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