diff --git a/optimum/onnxruntime/quantization.py b/optimum/onnxruntime/quantization.py index 08d7451b9f9..9677fbd912c 100644 --- a/optimum/onnxruntime/quantization.py +++ b/optimum/onnxruntime/quantization.py @@ -358,7 +358,7 @@ def quantize( if has_subgraphs: if quantization_config.is_static: raise NotImplementedError("Static quantization is currently not supported for models with subgraphs.") - if parse(ort_version) >= Version("1.16.0"): + if parse(ort_version) == Version("1.16.0"): raise ValueError( "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" )