Adds FSMT to LM head AutoModel (#7312)

2020-09-22 06:35:51 -04:00
parent e2964b8a19
commit e46108817e
1 changed files with 1 additions and 0 deletions
@@ -258,6 +258,7 @@ MODEL_WITH_LM_HEAD_MAPPING = OrderedDict(
        (CamembertConfig, CamembertForMaskedLM),
        (XLMRobertaConfig, XLMRobertaForMaskedLM),
        (MarianConfig, MarianMTModel),
+        (FSMTConfig, FSMTForConditionalGeneration),
        (BartConfig, BartForConditionalGeneration),
        (LongformerConfig, LongformerForMaskedLM),
        (RobertaConfig, RobertaForMaskedLM),