update phi default onnx opset

huggingface · Mar 21, 2024 · a809853 · a809853
1 parent 68f2c4e
commit a809853
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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