feat: added from_config, also pass additional kwargs from config to model

modeling_bert.py

@@ -328,13 +328,17 @@ class BertPreTrainedModel(nn.Module):
                 (ex: num_labels for BertForSequenceClassification)
         """
         # Instantiate model.
-        model = cls(config) #cls(config, *inputs, **kwargs)
+        model = cls(config, *inputs, **kwargs)
         load_return = model.load_state_dict(
             remap_state_dict(state_dict_from_pretrained(model_name), config), strict=False
         )
         logger.info(load_return)
         return model
 
+    @classmethod
+    def from_config(cls, config, *inputs, **kwargs):
+        model = cls(config, *inputs, **kwargs)
+        return model
 
 class BertModel(BertPreTrainedModel):
     def __init__(self, config: JinaBertConfig, add_pooling_layer=True):
@@ -523,6 +527,10 @@ class BertForPreTraining(BertPreTrainedModel):
             seq_relationship_logits=seq_relationship_score,
         )
 
+    @classmethod
+    def _from_config(cls, config, **kwargs):
+        pass
+
 
 def remap_state_dict(state_dict, config: PretrainedConfig):
     """