Update BertForJointParsing.py

BertForJointParsing.py

@@ -40,7 +40,10 @@ class ModuleRef:
         return self.module(*args, **kwargs)
 
 class BertForJointParsing(BertPreTrainedModel):
-    _tied_weights_keys = ["predictions.decoder.bias", "cls.predictions.decoder.weight"]
+    tied_weights_keys = {
+        "cls.predictions.decoder.weight": "bert.embeddings.word_embeddings.weight",
+        "cls.predictions.decoder.bias": "cls.predictions.bias",
+    }
 
     def __init__(self, config, do_syntax=None, do_ner=None, do_prefix=None, do_lex=None, do_morph=None, syntax_head_size=64):
         super().__init__(config)