[fix] add bart to LM_MAPPING (#6099)

2020-07-28 18:29:18 -04:00
parent 5abe50381a
commit 40796c5801
1 changed files with 1 additions and 0 deletions
@@ -264,6 +264,7 @@ MODEL_FOR_MASKED_LM_MAPPING = OrderedDict(
    [
        (DistilBertConfig, DistilBertForMaskedLM),
        (AlbertConfig, AlbertForMaskedLM),
+        (BartConfig, BartForConditionalGeneration),
        (CamembertConfig, CamembertForMaskedLM),
        (XLMRobertaConfig, XLMRobertaForMaskedLM),
        (LongformerConfig, LongformerForMaskedLM),