Fixed the bugs caused by the version update of transformers.

QiaoZiqing · QiaoZiqing · commit 90eb498fc190 · 2021-12-06T14:45:49.000+08:00
diff --git a/opennre/encoder/bert_encoder.py b/opennre/encoder/bert_encoder.py
@@ -29,6 +29,7 @@ def forward(self, token, att_mask, pos1, pos2):
             (B, H), representations for sentences
         """
         _, x = self.bert(token, attention_mask=att_mask, return_dict=False)
+        #return_dict=fault is set to adapt to the new version of transformers
         return x
 
     def tokenize(self, item):
