From 88395261d8fa7fb0b18764cf346b985f586f9b2f Mon Sep 17 00:00:00 2001
From: Stephan Auerhahn <stephan@stability.ai>
Date: Thu, 10 Aug 2023 12:45:37 -0700
Subject: [PATCH] update helpers

---
 scripts/demo/streamlit_helpers.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/scripts/demo/streamlit_helpers.py b/scripts/demo/streamlit_helpers.py
index 59cd27b..b37e2eb 100644
--- a/scripts/demo/streamlit_helpers.py
+++ b/scripts/demo/streamlit_helpers.py
@@ -24,14 +24,14 @@ from sgm.inference.helpers import embed_watermark, CudaModelManager
 
 
 @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,