riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 11:11:05 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Adds FSMT to LM head AutoModel (#7312)

Browse Source
This commit is contained in:
Lysandre Debut 2020-09-22 12:35:51 +02:00 committed by GitHub
parent e2964b8a19
commit e46108817e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/modeling_auto.py
View File

@ -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),