CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

imports

Browse Source
This commit is contained in:
Haoming 2025-07-28 11:47:30 +08:00
parent 8be968b7c6
commit 923113bd4b
2 changed files with 15 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
modules/initialize.py
View File

@ -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")

3
webui.py
View File

@ -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()