fix opt

huggingface · Oct 24, 2024 · 6191617 · 6191617
1 parent c860ba9
commit 6191617
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/exporters/onnx/model_configs.py b/optimum/exporters/onnx/model_configs.py
@@ -269,7 +269,7 @@ class GPTNeoXOnnxConfig(TextDecoderWithPositionIdsOnnxConfig):
 
 class OPTOnnxConfig(TextDecoderOnnxConfig):
     # OPT does not require position_ids input.
-    DEFAULT_ONNX_OPSET = 13
+    DEFAULT_ONNX_OPSET = 14  # uses SDPA in Transformers, hence opset>=14.
     NORMALIZED_CONFIG_CLASS = NormalizedTextConfig