fix: revert changes to configuration_bert

configuration_bert.py

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