update helpers

Stability-AI · Aug 10, 2023 · 8839526 · 8839526
1 parent b3866d1
commit 8839526
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scripts/demo/streamlit_helpers.py b/scripts/demo/streamlit_helpers.py
@@ -24,14 +24,14 @@
 
 
 @st.cache_resource()
-def init_st(spec: SamplingSpec, load_ckpt=True, load_filter=True) -> Dict[str, Any]:
+def init_st(
+    spec: SamplingSpec, load_ckpt=True, load_filter=True, lowvram_mode=True
+) -> Dict[str, Any]:
     state: Dict[str, Any] = dict()
     if not "model" in state:
         config = spec.config
         ckpt = spec.ckpt
 
-        lowvram_mode = st.checkbox("Low VRAM mode", value=False)
-
         if lowvram_mode:
             pipeline = SamplingPipeline(
                 model_spec=spec,