feat: added get_input_embeddings method to BertForPreTraining

modeling_bert.py

@@ -459,6 +459,9 @@ class BertForPreTraining(BertPreTrainedModel):
     def tie_weights(self):
         self.cls.predictions.decoder.weight = self.bert.embeddings.word_embeddings.weight
 
+    def get_input_embeddings(self):
+        return self.embeddings.word_embeddings
+
     def forward(
         self,
         input_ids,