fix the mlm longformer example by changing [MASK] to <mask> (#11559)

6a11e4c2 · Frederik Bode · GitHub · 1c86157d · 6a11e4c2
Unverified Commit 6a11e4c2 authored May 03, 2021 by Frederik Bode Committed by GitHub May 03, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/transformers/models/longformer/modeling_tf_longformer.py src/transformers/models/longformer/modeling_tf_longformer.py +1 -0

No files found.
--- a/src/transformers/models/longformer/modeling_tf_longformer.py
+++ b/src/transformers/models/longformer/modeling_tf_longformer.py
@@ -2082,6 +2082,7 @@ class TFLongformerForMaskedLM(TFLongformerPreTrainedModel, TFMaskedLanguageModel
        checkpoint=_CHECKPOINT_FOR_DOC,
        output_type=TFLongformerMaskedLMOutput,
        config_class=_CONFIG_FOR_DOC,
+        mask="<mask>",
    )
    def call(
        self,