Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

reorg imports for faster startup

Browse Source
This commit is contained in:
Vladimir Mandic
2024-04-07 16:29:33 -04:00
parent 01b9c4ddf0
commit 19db61a861
3 changed files with 17 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/loader.py
+7 -3
View File
@@ -37,13 +37,17 @@ if ".dev" in torch.__version__ or "+git" in torch.__version__:
torch.__version__ = re.search(r'[\d.]+[\d]', torch.__version__).group(0)
timer.startup.record("torch")
import transformers # pylint: disable=W0611,C0411
timer.startup.record("transformers")
import onnxruntime
onnxruntime.set_default_logger_severity(3)
timer.startup.record("onnx")
from modules.onnx_impl import initialize_olive # pylint: disable=ungrouped-imports
initialize_olive()
timer.startup.record("olive")
# moved to webui.py:initialize()
# from modules.onnx_impl import initialize_olive # pylint: disable=ungrouped-imports
# initialize_olive()
# timer.startup.record("olive")
from fastapi import FastAPI # pylint: disable=W0611,C0411
import gradio # pylint: disable=W0611,C0411