From d6f2b7899429496aae347229a2f0f522cbb5c38b Mon Sep 17 00:00:00 2001
From: Stephan Auerhahn <stephan@stability.ai>
Date: Thu, 10 Aug 2023 15:06:55 -0700
Subject: [PATCH] pass options into state2 init

---
 scripts/demo/sampling.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/scripts/demo/sampling.py b/scripts/demo/sampling.py
index 87d6feb..a18b21a 100644
--- a/scripts/demo/sampling.py
+++ b/scripts/demo/sampling.py
@@ -279,7 +279,9 @@ if __name__ == "__main__":
         st.write("**Refiner Options:**")
 
         specs2 = model_specs[version2]
-        state2 = init_st(specs2, load_filter=False)
+        state2 = init_st(
+            specs2, load_filter=False, use_fp16=use_fp16, enable_swap=enable_swap
+        )
         params2 = state2["params"]
 
         params2.img2img_strength = st.number_input(