fix: revert changes to configuration_bert
Browse files- configuration_bert.py +2 -2
configuration_bert.py
CHANGED
@@ -19,7 +19,7 @@ from collections import OrderedDict
|
|
19 |
from typing import Mapping
|
20 |
|
21 |
from transformers.configuration_utils import PretrainedConfig
|
22 |
-
from
|
23 |
from transformers.utils import logging
|
24 |
|
25 |
|
@@ -152,7 +152,7 @@ class JinaBertConfig(PretrainedConfig):
|
|
152 |
self.emb_pooler = emb_pooler
|
153 |
self.attn_implementation = attn_implementation
|
154 |
|
155 |
-
class JinaBertOnnxConfig(
|
156 |
@property
|
157 |
def inputs(self) -> Mapping[str, Mapping[int, str]]:
|
158 |
if self.task == "multiple-choice":
|
|
|
19 |
from typing import Mapping
|
20 |
|
21 |
from transformers.configuration_utils import PretrainedConfig
|
22 |
+
from transformers.onnx import OnnxConfig
|
23 |
from transformers.utils import logging
|
24 |
|
25 |
|
|
|
152 |
self.emb_pooler = emb_pooler
|
153 |
self.attn_implementation = attn_implementation
|
154 |
|
155 |
+
class JinaBertOnnxConfig(OnnxConfig):
|
156 |
@property
|
157 |
def inputs(self) -> Mapping[str, Mapping[int, str]]:
|
158 |
if self.task == "multiple-choice":
|