added --exit for notebooks

3 years ago · dc1ee79e96
parent 8a50371e4e
commit dc1ee79e96
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:])}")