mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 10:12:23 +06:00
fix the mlm longformer example by changing [MASK] to <mask> (#11559)
This commit is contained in:
parent
1c86157d9d
commit
6a11e4c2ad
@ -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,
|
||||||
|
Loading…
Reference in New Issue
Block a user