Skip to content

GitLab

Unverified Commit 6a11e4c2 authored by Frederik Bode's avatar Frederik Bode Committed by GitHub
Browse files

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

parent 1c86157d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/models/longformer/modeling_tf_longformer.py
View file @ 6a11e4c2
...@@ -2082,6 +2082,7 @@ class TFLongformerForMaskedLM(TFLongformerPreTrainedModel, TFMaskedLanguageModel ...@@ -2082,6 +2082,7 @@ class TFLongformerForMaskedLM(TFLongformerPreTrainedModel, TFMaskedLanguageModel
checkpoint=_CHECKPOINT_FOR_DOC, checkpoint=_CHECKPOINT_FOR_DOC,
output_type=TFLongformerMaskedLMOutput, output_type=TFLongformerMaskedLMOutput,
config_class=_CONFIG_FOR_DOC, config_class=_CONFIG_FOR_DOC,
mask="<mask>",
) )
def call( def call(
self, self,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment