We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 58aec63 commit 32725caCopy full SHA for 32725ca
optimum/exporters/openvino/convert.py
@@ -664,6 +664,9 @@ def export_from_model(
664
# Get the shapes to be used to generate dummy inputs
665
input_shapes = {}
666
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
670
input_shapes[input_name] = (
671
kwargs_shapes[input_name] if input_name in kwargs_shapes else DEFAULT_DUMMY_SHAPES[input_name]
672
)
0 commit comments