diff --git a/modules/initialize.py b/modules/initialize.py index 8251bce8..f7313ff4 100644 --- a/modules/initialize.py +++ b/modules/initialize.py @@ -12,10 +12,6 @@ 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()) - from modules_forge.initialization import initialize_forge - initialize_forge() - startup_timer.record("initialize forge") - import torch # noqa: F401 startup_timer.record("import torch") import pytorch_lightning # noqa: F401 diff --git a/webui.py b/webui.py index 2c417168..e15cf452 100644 --- a/webui.py +++ b/webui.py @@ -7,9 +7,13 @@ from modules import timer from modules import initialize_util from modules import initialize +from modules_forge.initialization import initialize_forge + startup_timer = timer.startup_timer startup_timer.record("launcher") +initialize_forge() + initialize.imports() initialize.check_versions()