Skip to content

Commit 32725ca

Browse files
committed
Remove height and width from default shapes pre-configuration
1 parent 58aec63 commit 32725ca

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

optimum/exporters/openvino/convert.py

+3
Original file line numberDiff line numberDiff line change
@@ -664,6 +664,9 @@ def export_from_model(
664664
# Get the shapes to be used to generate dummy inputs
665665
input_shapes = {}
666666
for input_name in DEFAULT_DUMMY_SHAPES.keys():
667+
if input_name in ["height", "width"]:
668+
# use H and W from generator defaults
669+
continue
667670
input_shapes[input_name] = (
668671
kwargs_shapes[input_name] if input_name in kwargs_shapes else DEFAULT_DUMMY_SHAPES[input_name]
669672
)

0 commit comments

Comments
 (0)