From b717eb7e56a4e620e77a2225e80223c89cb4f0d1 Mon Sep 17 00:00:00 2001
From: AUTOMATIC1111 <16777216c@gmail.com>
Date: Thu, 13 Jul 2023 08:29:37 +0300
Subject: [PATCH] mute unneeded SDXL imports for tests too

---
 modules/launch_utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/modules/launch_utils.py b/modules/launch_utils.py
index 4f48f3a13..56b972d51 100644
--- 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: