raise atol for MT5OnnxConfig (#18560)

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
huggingface · Aug 10, 2022 · 9a9a525 · 9a9a525
1 parent f62cb83
commit 9a9a525
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/transformers/models/mt5/configuration_mt5.py b/src/transformers/models/mt5/configuration_mt5.py
@@ -147,9 +147,9 @@ def num_hidden_layers(self):
         return self.num_layers
 
 
-# Copied from transformers.models.t5.configuration_t5.T5OnnxConfig
 class MT5OnnxConfig(OnnxSeq2SeqConfigWithPast):
     @property
+    # Copied from transformers.models.t5.configuration_t5.T5OnnxConfig.inputs
     def inputs(self) -> Mapping[str, Mapping[int, str]]:
         common_inputs = {
             "input_ids": {0: "batch", 1: "encoder_sequence"},
@@ -169,5 +169,10 @@ def inputs(self) -> Mapping[str, Mapping[int, str]]:
         return common_inputs
 
     @property
+    # Copied from transformers.models.t5.configuration_t5.T5OnnxConfig.default_onnx_opset
     def default_onnx_opset(self) -> int:
         return 13
+
+    @property
+    def atol_for_validation(self) -> float:
+        return 5e-4