Skip to content

GitLab

Unverified Commit 40796c58 authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

[fix] add bart to LM_MAPPING (#6099)

parent 5abe5038
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/modeling_auto.py
View file @ 40796c58
...@@ -264,6 +264,7 @@ MODEL_FOR_MASKED_LM_MAPPING = OrderedDict( ...@@ -264,6 +264,7 @@ MODEL_FOR_MASKED_LM_MAPPING = OrderedDict(
[ [
(DistilBertConfig, DistilBertForMaskedLM), (DistilBertConfig, DistilBertForMaskedLM),
(AlbertConfig, AlbertForMaskedLM), (AlbertConfig, AlbertForMaskedLM),
(BartConfig, BartForConditionalGeneration),
(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