From 45168178dcb6bce14c0d65ccfaebf4cfa74bd899 Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Fri, 26 Jun 2026 02:13:39 +0200 Subject: [PATCH] ModelZoo ONNX backend: log configured top provider (backend) --- src/eynollah/model_zoo/model_zoo.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/eynollah/model_zoo/model_zoo.py b/src/eynollah/model_zoo/model_zoo.py index 232a28a..d6b5e38 100644 --- a/src/eynollah/model_zoo/model_zoo.py +++ b/src/eynollah/model_zoo/model_zoo.py @@ -351,10 +351,15 @@ class EynollahModelZoo: # 'trt_timing_cache_enable': True, # ... })] + providers + provider0 = providers[0] + if isinstance(provider0, tuple): + provider0 = provider0[0] + self.logger.info("using %s with ONNX provider %s for model %s", + "GPU %d" % gpu if gpu >= 0 else "CPU", + provider0[:-17], model_category) model = ort.InferenceSession( model_path, providers=providers) - # FIXME: notify about selected provider/device model_inputs = [model_input.name for model_input in model.get_inputs()] model_outputs = [model_output.name