fix the mlm longformer example by changing [MASK] to <mask> (huggingf…

…ace#11559)
Iwontbecreative · Jul 15, 2021 · 62ab558 · 62ab558
1 parent 7a5d822
commit 62ab558
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/transformers/models/longformer/modeling_tf_longformer.py b/src/transformers/models/longformer/modeling_tf_longformer.py
@@ -2082,6 +2082,7 @@ def get_prefix_bias_name(self):
         checkpoint=_CHECKPOINT_FOR_DOC,
         output_type=TFLongformerMaskedLMOutput,
         config_class=_CONFIG_FOR_DOC,
+        mask="<mask>",
     )
     def call(
         self,