diff --git a/launch.py b/launch.py index b2732a4d..ad41ee41 100644 --- 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:])}")