Markus28 commited on
Commit
45b2292
1 Parent(s): 80472cb

added config_class and base_model_prefix

Browse files
Files changed (1) hide show
  1. 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':