added --exit for notebooks

2025-02-23 15:59:50 +08:00 · 2022-09-29 18:33:29 +03:00 · 2022-09-29 18:33:29 +03:00 · dc1ee79e96
commit dc1ee79e96
parent 8a50371e4e
1 changed files with 3 additions and 0 deletions
--- a/launch.py
+++ b/launch.py
@ -130,6 +130,9 @@ run_pip(f"install -r {requirements_file}", "requirements for Web UI")

 sys.argv += args

+if "--exit" in args:
+    print("Exiting because of --exit argument")
+    exit(0)

 def start_webui():
    print(f"Launching Web UI with arguments: {' '.join(sys.argv[1:])}")