From a809853d40423e54bff19da1fc7efbab2a35d7bb Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Thu, 21 Mar 2024 16:42:46 +0100 Subject: [PATCH] update phi default onnx opset --- optimum/exporters/onnx/model_configs.py | 1 + 1 file changed, 1 insertion(+) diff --git a/optimum/exporters/onnx/model_configs.py b/optimum/exporters/onnx/model_configs.py index b43ca23969..cd9d54eeca 100644 --- a/optimum/exporters/onnx/model_configs.py +++ b/optimum/exporters/onnx/model_configs.py @@ -252,6 +252,7 @@ class GemmaOnnxConfig(LlamaOnnxConfig): class PhiOnnxConfig(TextDecoderWithPositionIdsOnnxConfig): + DEFAULT_ONNX_OPSET = 14 # Phi now uses F.scaled_dot_product_attention by default for torch>=2.1.1. NORMALIZED_CONFIG_CLASS = NormalizedTextConfig