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(
(MPNetConfig, MPNetForMaskedLM),
(TapasConfig, TapasForMaskedLM),
(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