added config_class and base_model_prefix
Browse files- modeling_bert.py +3 -1
modeling_bert.py
CHANGED
|
@@ -298,7 +298,9 @@ class BertPreTrainedModel(PreTrainedModel):
|
|
| 298 |
"""An abstract class to handle weights initialization and
|
| 299 |
a simple interface for dowloading and loading pretrained models.
|
| 300 |
"""
|
| 301 |
-
|
|
|
|
|
|
|
| 302 |
def __init__(self, config, *inputs, **kwargs):
|
| 303 |
super().__init__()
|
| 304 |
if not config.__class__.__name__ == 'JinaBertConfig':
|
|
|
|
| 298 |
"""An abstract class to handle weights initialization and
|
| 299 |
a simple interface for dowloading and loading pretrained models.
|
| 300 |
"""
|
| 301 |
+
config_class = JinaBertConfig
|
| 302 |
+
base_model_prefix = "bert"
|
| 303 |
+
|
| 304 |
def __init__(self, config, *inputs, **kwargs):
|
| 305 |
super().__init__()
|
| 306 |
if not config.__class__.__name__ == 'JinaBertConfig':
|