Update optimum/onnxruntime/quantization.py

Co-authored-by: fxmarty <[email protected]>
huggingface · Sep 21, 2023 · ac550be · ac550be
1 parent b343ce4
commit ac550be
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"
                 )