Add DeBERTa to MODEL_FOR_PRETRAINING_MAPPING (#10668)
* add deberta to pretraining mapping * add deberta_v2 to PRETRAINING_MAPPING
This commit is contained in:
parent
9fbb4cdc80
commit
a01ea31b5c
|
@ -394,6 +394,8 @@ MODEL_FOR_PRETRAINING_MAPPING = OrderedDict(
|
|||
(MPNetConfig, MPNetForMaskedLM),
|
||||
(TapasConfig, TapasForMaskedLM),
|
||||
(IBertConfig, IBertForMaskedLM),
|
||||
(DebertaConfig, DebertaForMaskedLM),
|
||||
(DebertaV2Config, DebertaV2ForMaskedLM),
|
||||
]
|
||||
)
|
||||
|
||||
|
|
Loading…
Reference in New Issue