From ac550be1365d4e15a7edf117b0310ea7c5535265 Mon Sep 17 00:00:00 2001 From: Ella Charlaix <80481427+echarlaix@users.noreply.github.com> Date: Thu, 21 Sep 2023 16:10:41 +0200 Subject: [PATCH] Update optimum/onnxruntime/quantization.py Co-authored-by: fxmarty <9808326+fxmarty@users.noreply.github.com> --- optimum/onnxruntime/quantization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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" )