Skip to content

Commit

Permalink
Update optimum/onnxruntime/quantization.py
Browse files Browse the repository at this point in the history
Co-authored-by: fxmarty <[email protected]>
  • Loading branch information
echarlaix and fxmarty authored Sep 21, 2023
1 parent 3bea1a2 commit b343ce4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion optimum/onnxruntime/quantization.py
Original file line number Diff line number Diff line change
Expand Up @@ -360,7 +360,7 @@ def quantize(
raise NotImplementedError("Static quantization is currently not supported for models with subgraphs.")
if parse(ort_version) >= Version("1.16.0"):
raise ValueError(
"Onnxruntime version v1.16.0 not compatible with quantization for models with subgraphs, please downgrade to an earlier version."
"ONNX Runtime version v1.16.0 not compatible with quantization for models with subgraphs, please downgrade to 1.15.1 or upgrade to a higher version. Reference: https://github.com/microsoft/onnxruntime/pull/17651"
)

quantizer_factory = QDQQuantizer if use_qdq else ONNXQuantizer
Expand Down

0 comments on commit b343ce4

Please sign in to comment.