fix: fixed get_input_embeddings method

modeling_bert.py

@@ -460,7 +460,7 @@ class BertForPreTraining(BertPreTrainedModel):
         self.cls.predictions.decoder.weight = self.bert.embeddings.word_embeddings.weight
 
     def get_input_embeddings(self):
-        return self.embeddings.word_embeddings
+        return self.bert.embeddings.word_embeddings
 
     def forward(
         self,