mute unneeded SDXL imports for tests too

AUTOMATIC1111 · Jul 13, 2023 · b717eb7 · b717eb7
1 parent a04c955
commit b717eb7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/modules/launch_utils.py b/modules/launch_utils.py
@@ -334,8 +334,6 @@ def prepare_environment():
     if args.update_all_extensions:
         git_pull_recursive(extensions_dir)
 
-    mute_sdxl_imports()
-
     if "--exit" in sys.argv:
         print("Exiting because of --exit argument")
         exit(0)
@@ -357,6 +355,8 @@ def configure_for_tests():
 
 
 def start():
+    mute_sdxl_imports()
+
     print(f"Launching {'API server' if '--nowebui' in sys.argv else 'Web UI'} with arguments: {' '.join(sys.argv[1:])}")
     import webui
     if '--nowebui' in sys.argv: