Unverified Commit a01ea31b authored by jeswan's avatar jeswan Committed by GitHub
Browse files

Add DeBERTa to MODEL_FOR_PRETRAINING_MAPPING (#10668)

* add deberta to pretraining mapping

* add deberta_v2 to PRETRAINING_MAPPING
parent 9fbb4cdc
...@@ -394,6 +394,8 @@ MODEL_FOR_PRETRAINING_MAPPING = OrderedDict( ...@@ -394,6 +394,8 @@ MODEL_FOR_PRETRAINING_MAPPING = OrderedDict(
(MPNetConfig, MPNetForMaskedLM), (MPNetConfig, MPNetForMaskedLM),
(TapasConfig, TapasForMaskedLM), (TapasConfig, TapasForMaskedLM),
(IBertConfig, IBertForMaskedLM), (IBertConfig, IBertForMaskedLM),
(DebertaConfig, DebertaForMaskedLM),
(DebertaV2Config, DebertaV2ForMaskedLM),
] ]
) )
......
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