use OnnxConfig

huggingface · Dec 23, 2021 · d1be26d · d1be26d
1 parent c1de9ad
commit d1be26d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/transformers/models/ibert/configuration_ibert.py b/src/transformers/models/ibert/configuration_ibert.py
@@ -18,7 +18,7 @@
 from collections import OrderedDict
 from typing import Mapping
 
-from transformers.onnx import OnnxConfigWithPast
+from transformers.onnx import OnnxConfig
 
 from ...configuration_utils import PretrainedConfig
 from ...utils import logging
@@ -127,12 +127,12 @@ def __init__(
         self.force_dequant = force_dequant
 
 
-class IBertOnnxConfig(OnnxConfigWithPast):
+class IBertOnnxConfig(OnnxConfig):
     @property
     def inputs(self) -> Mapping[str, Mapping[int, str]]:
         return OrderedDict(
             [
                 ("input_ids", {0: "batch", 1: "sequence"}),
                 ("attention_mask", {0: "batch", 1: "sequence"}),
             ]
-        )
+        )