lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-30 19:21:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

early initilize

Browse Source
This commit is contained in:
lllyasviel
2024-01-31 15:27:13 -08:00
parent 9aa67d5578
commit 66187d97fb
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/initialize.py
View File

@@ -12,10 +12,6 @@ def imports():
logging.getLogger("torch.distributed.nn").setLevel(logging.ERROR) # sshh... 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()) logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())
from modules_forge.initialization import initialize_forge
initialize_forge()
startup_timer.record("initialize forge")
import torch # noqa: F401 import torch # noqa: F401
startup_timer.record("import torch") startup_timer.record("import torch")
import pytorch_lightning # noqa: F401 import pytorch_lightning # noqa: F401

4
webui.py
View File

@@ -7,9 +7,13 @@ from modules import timer
from modules import initialize_util from modules import initialize_util
from modules import initialize from modules import initialize
from modules_forge.initialization import initialize_forge
startup_timer = timer.startup_timer startup_timer = timer.startup_timer
startup_timer.record("launcher") startup_timer.record("launcher")
initialize_forge()
initialize.imports() initialize.imports()
initialize.check_versions() initialize.check_versions()