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':
|