Markus28 commited on
Commit
18eed80
1 Parent(s): eeb05a3

feat: fixed _from_config

Browse files
Files changed (1) hide show
  1. modeling_bert.py +4 -4
modeling_bert.py CHANGED
@@ -335,6 +335,10 @@ class BertPreTrainedModel(nn.Module):
335
  logger.info(load_return)
336
  return model
337
 
 
 
 
 
338
 
339
  class BertModel(BertPreTrainedModel):
340
  def __init__(self, config: JinaBertConfig, add_pooling_layer=True):
@@ -523,10 +527,6 @@ class BertForPreTraining(BertPreTrainedModel):
523
  seq_relationship_logits=seq_relationship_score,
524
  )
525
 
526
- @classmethod
527
- def _from_config(cls, config, **kwargs):
528
- pass
529
-
530
 
531
  def remap_state_dict(state_dict, config: PretrainedConfig):
532
  """
 
335
  logger.info(load_return)
336
  return model
337
 
338
+ @classmethod
339
+ def _from_config(cls, config, **kwargs):
340
+ return cls(config, **kwargs)
341
+
342
 
343
  class BertModel(BertPreTrainedModel):
344
  def __init__(self, config: JinaBertConfig, add_pooling_layer=True):
 
527
  seq_relationship_logits=seq_relationship_score,
528
  )
529
 
 
 
 
 
530
 
531
  def remap_state_dict(state_dict, config: PretrainedConfig):
532
  """