jinaai
/

jina-bert-flash-implementation

Markus28 commited on Feb 22

Commit

7e06371

•

1 Parent(s): bb281f0

fix: fixed get_input_embeddings method

Files changed (1) hide show

modeling_bert.py CHANGED Viewed

@@ -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
     def forward(
         self,

         self.cls.predictions.decoder.weight = self.bert.embeddings.word_embeddings.weight
     def get_input_embeddings(self):
+        return self.bert.embeddings.word_embeddings
     def forward(
         self,