From 32725ca0406592f3357c4253e0e0bd90ae6c2830 Mon Sep 17 00:00:00 2001 From: Maxim Vafin Date: Wed, 8 Jan 2025 17:51:46 +0100 Subject: [PATCH] Remove height and width from default shapes pre-configuration --- optimum/exporters/openvino/convert.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/optimum/exporters/openvino/convert.py b/optimum/exporters/openvino/convert.py index 04d4944e0..61568c8a7 100644 --- a/optimum/exporters/openvino/convert.py +++ b/optimum/exporters/openvino/convert.py @@ -664,6 +664,9 @@ def export_from_model( # Get the shapes to be used to generate dummy inputs input_shapes = {} for input_name in DEFAULT_DUMMY_SHAPES.keys(): + if input_name in ["height", "width"]: + # use H and W from generator defaults + continue input_shapes[input_name] = ( kwargs_shapes[input_name] if input_name in kwargs_shapes else DEFAULT_DUMMY_SHAPES[input_name] )