Update BertForJointParsing.py

BertForJointParsing.py

@@ -314,6 +314,7 @@ def lex_parse_logits(input_ids: List[List[int]], sentences: List[str], tokenizer
 
     special_toks = tokenizer.all_special_tokens
     special_toks.remove(tokenizer.unk_token)
+    special_toks.remove(tokenizer.mask_token)
     for batch_idx in range(len(sentences)):
         intermediate_ret = []
         tokens = tokenizer.convert_ids_to_tokens(input_ids[batch_idx])