Skip to content

GitLab

Unverified Commit 2390c16f authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

add mbart to automodel for masked lm (#9673)

parent b39bd763
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/models/auto/modeling_auto.py
View file @ 2390c16f
...@@ -411,6 +411,7 @@ MODEL_FOR_MASKED_LM_MAPPING = OrderedDict( ...@@ -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),
......
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