Update configuration_bert.py

configuration_bert.py

@@ -19,7 +19,7 @@ from collections import OrderedDict
 from typing import Mapping
 
 from transformers.configuration_utils import PretrainedConfig
-from transformers.onnx import OnnxConfig
+from optimum.exporters.onnx.model_configs import OnnxConfig, BertOnnxConfig
 from transformers.utils import logging
 
 
@@ -152,7 +152,8 @@ class JinaBertConfig(PretrainedConfig):
         self.emb_pooler = emb_pooler
         self.attn_implementation = attn_implementation
 
-class JinaBertOnnxConfig(OnnxConfig):
+class JinaBertOnnxConfig(BertOnnxConfig):
+    
     @property
     def inputs(self) -> Mapping[str, Mapping[int, str]]:
         if self.task == "multiple-choice":
@@ -163,6 +164,5 @@ class JinaBertOnnxConfig(OnnxConfig):
             [
                 ("input_ids", dynamic_axis),
                 ("attention_mask", dynamic_axis),
-                ("token_type_ids", dynamic_axis),
             ]
         )