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

add onnx initialization log

Browse Source
This commit is contained in:
Seunghoon Lee
2024-02-06 18:20:23 +09:00
parent cdc29fafce
commit 70a15defc5
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/onnx_impl/__init__.py
+5 -1
View File
@@ -205,9 +205,11 @@ def initialize():
if initialized:
return
from installer import log
from modules import devices
from modules.paths import models_path
from .execution_providers import ExecutionProvider, TORCH_DEVICE_TO_EP
from modules.shared import opts
from .execution_providers import ExecutionProvider, TORCH_DEVICE_TO_EP, available_execution_providers
onnx_dir = os.path.join(models_path, "ONNX")
if not os.path.isdir(onnx_dir):
@@ -250,6 +252,8 @@ def initialize():
optimum.onnxruntime.modeling_diffusion._ORTDiffusionModelPart.to = ORTDiffusionModelPart_to # pylint: disable=protected-access
log.info(f'ONNX: selected={opts.onnx_execution_provider}, available={available_execution_providers}')
initialized = True