fix t5

huggingface · Oct 23, 2024 · f0a4b7a · f0a4b7a
1 parent a4c3aa6
commit f0a4b7a
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
@@ -480,7 +480,7 @@ def generate(self, input_name: str, framework: str = "pt", int_dtype: str = "int
 
 
 class T5OnnxConfig(TextSeq2SeqOnnxConfig):
-    DEFAULT_ONNX_OPSET = 13
+    DEFAULT_ONNX_OPSET = 14  # T5 uses aten::triu that requires opset>=14
     DUMMY_INPUT_GENERATOR_CLASSES = TextSeq2SeqOnnxConfig.DUMMY_INPUT_GENERATOR_CLASSES[:-1] + (
         T5DummySeq2SeqPastKeyValuesGenerator,
     )