From 08946067acc881680eddd18c58495f44a5f03fe1 Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Fri, 12 Jun 2026 14:54:51 +0200 Subject: [PATCH] ModelZoo ONNX backend: handle multiple inputs, too --- src/eynollah/model_zoo/model_zoo.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/eynollah/model_zoo/model_zoo.py b/src/eynollah/model_zoo/model_zoo.py index 51ce909..c66c349 100644 --- a/src/eynollah/model_zoo/model_zoo.py +++ b/src/eynollah/model_zoo/model_zoo.py @@ -313,6 +313,7 @@ class EynollahModelZoo: # 'arena_extend_strategy': 'kNextPowerOfTwo', 'gpu_mem_limit': MODEL_VRAM_LIMITS[model_category] * 1024 * 1024, # 'cudnn_conv_algo_search': 'EXHAUSTIVE', + #'cudnn_conv_use_max_workspace': 0, # 'do_copy_in_default_stream': True, # ... })] + providers @@ -351,7 +352,12 @@ class EynollahModelZoo: outputs = outputs[0] return outputs model.predict_on_batch = predict_onnx - model.input_shape = model.get_inputs()[0].shape + input_spec = model.get_inputs() + input_spec = [i.shape for i in input_spec] + if len(input_spec) > 1: + model.input_shape = tuple(input_spec) + else: + model.input_shape = input_spec[0] return model