From 70a15defc5310c00eefa21630dddf58028093c13 Mon Sep 17 00:00:00 2001 From: Seunghoon Lee Date: Tue, 6 Feb 2024 18:20:23 +0900 Subject: [PATCH] add onnx initialization log --- modules/onnx_impl/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/modules/onnx_impl/__init__.py b/modules/onnx_impl/__init__.py index a6ab864f0..0a819c3aa 100644 --- a/modules/onnx_impl/__init__.py +++ b/modules/onnx_impl/__init__.py @@ -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