[fix] add bart to LM_MAPPING (#6099)

40796c58 · Sam Shleifer · GitHub · 5abe5038 · 40796c58
Unverified Commit 40796c58 authored Jul 28, 2020 by Sam Shleifer Committed by GitHub Jul 28, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/transformers/modeling_auto.py src/transformers/modeling_auto.py +1 -0

No files found.
--- a/src/transformers/modeling_auto.py
+++ b/src/transformers/modeling_auto.py
@@ -264,6 +264,7 @@ MODEL_FOR_MASKED_LM_MAPPING = OrderedDict(
    [
        (DistilBertConfig, DistilBertForMaskedLM),
        (AlbertConfig, AlbertForMaskedLM),
+        (BartConfig, BartForConditionalGeneration),
        (CamembertConfig, CamembertForMaskedLM),
        (XLMRobertaConfig, XLMRobertaForMaskedLM),
        (LongformerConfig, LongformerForMaskedLM),