45
45
AutoModelForMaskedImageModeling ,
46
46
AutoModelForMaskedLM ,
47
47
AutoModelForMultipleChoice ,
48
- AutoModelForNextSentencePrediction ,
49
48
AutoModelForQuestionAnswering ,
50
49
AutoModelForSeq2SeqLM ,
51
50
AutoModelForSequenceClassification ,
57
56
TFAutoModelForCausalLM ,
58
57
TFAutoModelForMaskedLM ,
59
58
TFAutoModelForMultipleChoice ,
60
- TFAutoModelForNextSentencePrediction ,
61
59
TFAutoModelForQuestionAnswering ,
62
60
TFAutoModelForSeq2SeqLM ,
63
61
TFAutoModelForSequenceClassification ,
@@ -112,7 +110,6 @@ class FeaturesManager:
112
110
"question-answering" : AutoModelForQuestionAnswering ,
113
111
"image-classification" : AutoModelForImageClassification ,
114
112
"masked-im" : AutoModelForMaskedImageModeling ,
115
- "next-sentence-prediction" : AutoModelForNextSentencePrediction ,
116
113
}
117
114
if is_tf_available ():
118
115
_TASKS_TO_TF_AUTOMODELS = {
@@ -124,7 +121,6 @@ class FeaturesManager:
124
121
"token-classification" : TFAutoModelForTokenClassification ,
125
122
"multiple-choice" : TFAutoModelForMultipleChoice ,
126
123
"question-answering" : TFAutoModelForQuestionAnswering ,
127
- "next-sentence-prediction" : TFAutoModelForNextSentencePrediction ,
128
124
}
129
125
130
126
# Set of model topologies we support associated to the features supported by each topology and the factory
@@ -159,7 +155,6 @@ class FeaturesManager:
159
155
"multiple-choice" ,
160
156
"token-classification" ,
161
157
"question-answering" ,
162
- "next-sentence-prediction" ,
163
158
onnx_config_cls = BertOnnxConfig ,
164
159
),
165
160
"big-bird" : supported_features_mapping (
@@ -326,7 +321,6 @@ class FeaturesManager:
326
321
"mobilebert" : supported_features_mapping (
327
322
"default" ,
328
323
"masked-lm" ,
329
- "next-sentence-prediction" ,
330
324
"sequence-classification" ,
331
325
"multiple-choice" ,
332
326
"token-classification" ,
0 commit comments