Add auto next sentence prediction (#8432)

* Add auto next sentence prediction

* Fix style

* Add mobilebert next sentence prediction
This commit is contained in:
Julien Plu
2020-11-10 17:11:48 +01:00
committed by GitHub
parent c314b1fd3b
commit 8551a99232
2 changed files with 107 additions and 2 deletions

View File

@@ -87,10 +87,8 @@ IGNORE_NON_AUTO_CONFIGURED = [
"RagSequenceForGeneration",
"RagTokenForGeneration",
"T5Stack",
"TFBertForNextSentencePrediction",
"TFFunnelBaseModel",
"TFGPT2DoubleHeadsModel",
"TFMobileBertForNextSentencePrediction",
"TFOpenAIGPTDoubleHeadsModel",
"XLMForQuestionAnswering",
"XLMProphetNetDecoder",