Modifies sys.argv depending on 'IS_SHARED_UI' env value + GPU presence

Dockerfile

@@ -121,4 +121,4 @@ RUN chmod +x on_start.sh
 
 EXPOSE 7860
 
-CMD ["/opt/venv/bin/python", "run.py", "--listen", "--enable-insecure-extension-access", "--ui-config-file", "ui-config.json", "--ui-settings-file", "config.json", "--disable-console-progressbars", "--cors-allow-origins", "huggingface.co,hf.space", "--no-progressbar-hiding", "--enable-console-prompts", "--no-download-sd-model", "--api", "--skip-version-check"]
+CMD ["/opt/venv/bin/python", "run.py", "--listen", "--ui-config-file", "ui-config.json", "--ui-settings-file", "config.json", "--disable-console-progressbars", "--cors-allow-origins", "huggingface.co,hf.space", "--no-progressbar-hiding", "--enable-console-prompts", "--no-download-sd-model", "--api", "--skip-version-check"]

header_patch.py

@@ -1,5 +1,6 @@
         with gr.Box(visible=os.environ.get("SPACE_ID")):
-            if os.environ.get("SPACE_ID") and str(os.environ.get("IS_SHARED_UI", "") or "") not in ("", "0"):
+            is_shared_ui = str(os.environ.get("IS_SHARED_UI", "") or "").strip().lower() not in ("", "0", "false", "none", "no")
+            if is_shared_ui:
                 import torch
                 if not torch.cuda.is_available():
                     gr.HTML(f"""

run.py

@@ -13,23 +13,43 @@ def on_start():
 
 
 def start():
-    on_start()
-
     print("---------------")
     print(f"Launching {'API server' if '--nowebui' in sys.argv else 'Web UI'} with arguments: {' '.join(sys.argv[1:])}")
     print("---------------")
     import webui  # type: ignore  # noqa
-    if '--nowebui' in sys.argv:
+
+    if "--nowebui" in sys.argv:
         webui.api_only()
     else:
         webui.webui()
 
 
-if __name__ == "__main__":
-    import torch
+def set_options():
+    import torch  # type: ignore  # noqa
+
     if not torch.cuda.is_available():
-        sys.argv.extend(["--precision", "full", "--no-half", "--use-cpu", "SD", "BSRGAN", "ESRGAN", "SCUNet", "CodeFormer", "--all"])
+        # If no GPU is available, uninstall xformers and apply "--precision full --no-half --use-cpu all" to sys.argv.
+        os.system(f"{sys.executable} -m pip uninstall -y xformers")
+        sys.argv.extend(
+            [
+                "--precision",
+                "full",
+                "--no-half",
+                "--use-cpu",
+                "all",
+            ]
+        )
     else:
+        # Applies "--force-enable-xformers --xformers" to sys.argv when there's a GPU present.
         sys.argv.extend(["--force-enable-xformers", "--xformers"])
 
+    is_shared_ui = str(os.environ.get("IS_SHARED_UI", "") or "").strip().lower() not in ("", "0", "false", "none", "no")
+    if not is_shared_ui:
+        # Provide access to extensions only if IS_SHARED_UI isn't set.
+        sys.argv.extend(["--enable-insecure-extension-access"])
+
+
+if __name__ == "__main__":
+    set_options()
+    on_start()
     start()