Skip to content

GitLab

"examples/run_classifier.py" did not exist on "34bdc8b54fc6cd9d3877df43e23356cfd79cdfe9"
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
Hide 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(
(CamembertConfig, CamembertForMaskedLM),
(XLMRobertaConfig, XLMRobertaForMaskedLM),
(MarianConfig, MarianMTModel),
(FSMTConfig, FSMTForConditionalGeneration),
(BartConfig, BartForConditionalGeneration),
(LongformerConfig, LongformerForMaskedLM),
(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