pass options into state2 init

Stability-AI · Aug 10, 2023 · d6f2b78 · d6f2b78
1 parent cd81956
commit d6f2b78
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/scripts/demo/sampling.py b/scripts/demo/sampling.py
@@ -279,7 +279,9 @@ def apply_refiner(
         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(