From 923113bd4b2e0bda0ce3f8552def08bda72ca894 Mon Sep 17 00:00:00 2001
From: Haoming <hmstudy02@gmail.com>
Date: Mon, 28 Jul 2025 11:47:30 +0800
Subject: [PATCH] imports

---
 modules/initialize.py | 21 ++++++++++++---------
 webui.py              |  3 +++
 2 files changed, 15 insertions(+), 9 deletions(-)

diff --git a/modules/initialize.py b/modules/initialize.py
index 536b4959..1fa57910 100644
--- a/modules/initialize.py
+++ b/modules/initialize.py
@@ -7,18 +7,21 @@ import os
 from modules.timer import startup_timer
 
 
-def imports():
-    logging.getLogger("torch.distributed.nn").setLevel(logging.ERROR)  # sshh...
-    logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())
-
-    import torch  # noqa: F401
-    startup_timer.record("import torch")
-    import pytorch_lightning  # noqa: F401
-    startup_timer.record("import torch")
+def shush():
+    logging.getLogger("torch.distributed.nn").setLevel(logging.ERROR)
+    logging.getLogger("xformers").addFilter(lambda record: "triton" not in record.getMessage().lower())
     warnings.filterwarnings(action="ignore", category=DeprecationWarning, module="pytorch_lightning")
     warnings.filterwarnings(action="ignore", category=UserWarning, module="torchvision")
 
-    os.environ.setdefault('GRADIO_ANALYTICS_ENABLED', 'False')
+
+def imports():
+    import torch  # noqa: F401
+    import torchvision  # noqa: F401
+    import pytorch_lightning  # noqa: F401
+    startup_timer.record("import torch")
+
+    os.environ.setdefault("GRADIO_ANALYTICS_ENABLED", "False")
+
     import gradio  # noqa: F401
     startup_timer.record("import gradio")
 
diff --git a/webui.py b/webui.py
index 247612bc..cffb5e1c 100644
--- a/webui.py
+++ b/webui.py
@@ -18,7 +18,10 @@ from modules_forge import main_thread
 startup_timer = timer.startup_timer
 startup_timer.record("launcher")
 
+initialize.shush()
+
 initialize_forge()
+startup_timer.record("forge init")
 
 initialize.imports()