SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

add mbart to automodel for masked lm (#9673)

Browse Source
This commit is contained in:
Patrick von Platen
2021-01-19 15:19:11 +01:00
committed by GitHub
parent b39bd763e8
commit 2390c16fd2
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/models/auto/modeling_auto.py
View File

@@ -411,6 +411,7 @@ MODEL_FOR_MASKED_LM_MAPPING = OrderedDict(
(DistilBertConfig, DistilBertForMaskedLM), (DistilBertConfig, DistilBertForMaskedLM),
(AlbertConfig, AlbertForMaskedLM), (AlbertConfig, AlbertForMaskedLM),
(BartConfig, BartForConditionalGeneration), (BartConfig, BartForConditionalGeneration),
(MBartConfig, MBartForConditionalGeneration),
(CamembertConfig, CamembertForMaskedLM), (CamembertConfig, CamembertForMaskedLM),
(XLMRobertaConfig, XLMRobertaForMaskedLM), (XLMRobertaConfig, XLMRobertaForMaskedLM),
(LongformerConfig, LongformerForMaskedLM), (LongformerConfig, LongformerForMaskedLM),