mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
+7
-9
@@ -133,13 +133,6 @@ 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")
|
||||
|
||||
try:
|
||||
from modules.sd_hijack_triton import install as install_autotune_report # pylint: disable=ungrouped-imports
|
||||
install_autotune_report()
|
||||
except Exception as e:
|
||||
log.warning(f'Triton logging: {e}')
|
||||
timer.startup.record("triton")
|
||||
|
||||
try:
|
||||
import bitsandbytes # pylint: disable=unused-import
|
||||
_bnb = True
|
||||
@@ -201,8 +194,6 @@ except Exception as e:
|
||||
_onnx = False
|
||||
timer.startup.record("onnx")
|
||||
|
||||
timer.startup.record("fastapi")
|
||||
|
||||
import gradio # pylint: disable=W0611,C0411
|
||||
timer.startup.record("gradio")
|
||||
errors.install([gradio])
|
||||
@@ -237,6 +228,13 @@ diffusers.utils.import_utils._sdnq_available = True # pylint: disable=protected-
|
||||
diffusers.utils.import_utils._sdnq_version = sdnq.__version__ # pylint: disable=protected-access
|
||||
timer.startup.record("sdnq")
|
||||
|
||||
try:
|
||||
from modules.sd_hijack_triton import install as install_autotune_report # pylint: disable=ungrouped-imports
|
||||
install_autotune_report()
|
||||
except Exception as e:
|
||||
log.warning(f'Triton logging: {e}')
|
||||
timer.startup.record("triton")
|
||||
|
||||
try:
|
||||
import pillow_jxl # pylint: disable=W0611,C0411
|
||||
except Exception:
|
||||
|
||||
Reference in New Issue
Block a user