Skip to content

GitLab

Unverified Commit e4610881 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Adds FSMT to LM head AutoModel (#7312)

parent e2964b8a
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/modeling_auto.py
View file @ e4610881
...@@ -258,6 +258,7 @@ MODEL_WITH_LM_HEAD_MAPPING = OrderedDict( ...@@ -258,6 +258,7 @@ MODEL_WITH_LM_HEAD_MAPPING = OrderedDict(
(CamembertConfig, CamembertForMaskedLM), (CamembertConfig, CamembertForMaskedLM),
(XLMRobertaConfig, XLMRobertaForMaskedLM), (XLMRobertaConfig, XLMRobertaForMaskedLM),
(MarianConfig, MarianMTModel), (MarianConfig, MarianMTModel),
(FSMTConfig, FSMTForConditionalGeneration),
(BartConfig, BartForConditionalGeneration), (BartConfig, BartForConditionalGeneration),
(LongformerConfig, LongformerForMaskedLM), (LongformerConfig, LongformerForMaskedLM),
(RobertaConfig, RobertaForMaskedLM), (RobertaConfig, RobertaForMaskedLM),
......
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