Update BertForMorphTagging.py

BertForMorphTagging.py

@@ -177,6 +177,8 @@ def parse_logits(input_ids: List[List[int]], sentences: List[str], tokenizer: Be
     #       { pos: str, feats: dict, prefixes: List[str], suffix: str | bool, suffix_feats: dict | None}
     special_toks = tokenizer.all_special_tokens
     special_toks.remove(tokenizer.unk_token)
+    special_toks.remove(tokenizer.mask_token)
+    
     ret = []
     for sent_idx,sentence in enumerate(sentences):
         input_id_strs = tokenizer.convert_ids_to_tokens(input_ids[sent_idx])