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

Add DeBERTa to MODEL_FOR_PRETRAINING_MAPPING (#10668)

Browse Source 
* add deberta to pretraining mapping

* add deberta_v2 to PRETRAINING_MAPPING
This commit is contained in:
jeswan 2021-03-11 13:56:47 -05:00 committed by GitHub
parent 9fbb4cdc80
commit a01ea31b5c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/auto/modeling_auto.py
View File

@ -394,6 +394,8 @@ MODEL_FOR_PRETRAINING_MAPPING = OrderedDict(
(MPNetConfig, MPNetForMaskedLM),
(TapasConfig, TapasForMaskedLM),
(IBertConfig, IBertForMaskedLM),
(DebertaConfig, DebertaForMaskedLM),
(DebertaV2Config, DebertaV2ForMaskedLM),
]
)