alaeddine-13
commited on
Commit
·
33026dc
1
Parent(s):
b4f2b16
rename to jina bert in configuration file
Browse files- configuration_bert.py +2 -2
configuration_bert.py
CHANGED
@@ -69,7 +69,7 @@ BERT_PRETRAINED_CONFIG_ARCHIVE_MAP = {
|
|
69 |
}
|
70 |
|
71 |
|
72 |
-
class
|
73 |
r"""
|
74 |
This is the configuration class to store the configuration of a [`BertModel`] or a [`TFBertModel`]. It is used to
|
75 |
instantiate a BERT model according to the specified arguments, defining the model architecture. Instantiating a
|
@@ -187,7 +187,7 @@ class MyBertConfig(PretrainedConfig):
|
|
187 |
self.emb_pooler = emb_pooler
|
188 |
|
189 |
|
190 |
-
class
|
191 |
@property
|
192 |
def inputs(self) -> Mapping[str, Mapping[int, str]]:
|
193 |
if self.task == "multiple-choice":
|
|
|
69 |
}
|
70 |
|
71 |
|
72 |
+
class JinaBertConfig(PretrainedConfig):
|
73 |
r"""
|
74 |
This is the configuration class to store the configuration of a [`BertModel`] or a [`TFBertModel`]. It is used to
|
75 |
instantiate a BERT model according to the specified arguments, defining the model architecture. Instantiating a
|
|
|
187 |
self.emb_pooler = emb_pooler
|
188 |
|
189 |
|
190 |
+
class JinaBertOnnxConfig(OnnxConfig):
|
191 |
@property
|
192 |
def inputs(self) -> Mapping[str, Mapping[int, str]]:
|
193 |
if self.task == "multiple-choice":
|