Fix quality and repo consistency

This commit is contained in:
Sylvain Gugger
2022-05-09 11:14:36 -04:00
parent 05fc1766ff
commit 7783fa6bb3
4 changed files with 10 additions and 3 deletions

View File

@@ -68,7 +68,7 @@ Ready-made configurations include the following architectures:
- M2M100 - M2M100
- Marian - Marian
- mBART - mBART
- MobileBert - MobileBERT
- OpenAI GPT-2 - OpenAI GPT-2
- PLBart - PLBart
- RoBERTa - RoBERTa

View File

@@ -1793,12 +1793,12 @@ if is_tf_available():
"TFAutoModelForImageClassification", "TFAutoModelForImageClassification",
"TFAutoModelForMaskedLM", "TFAutoModelForMaskedLM",
"TFAutoModelForMultipleChoice", "TFAutoModelForMultipleChoice",
"TFAutoModelForNextSentencePrediction",
"TFAutoModelForPreTraining", "TFAutoModelForPreTraining",
"TFAutoModelForQuestionAnswering", "TFAutoModelForQuestionAnswering",
"TFAutoModelForSeq2SeqLM", "TFAutoModelForSeq2SeqLM",
"TFAutoModelForSequenceClassification", "TFAutoModelForSequenceClassification",
"TFAutoModelForSpeechSeq2Seq", "TFAutoModelForSpeechSeq2Seq",
"TFAutoModelForNextSentencePrediction",
"TFAutoModelForTableQuestionAnswering", "TFAutoModelForTableQuestionAnswering",
"TFAutoModelForTokenClassification", "TFAutoModelForTokenClassification",
"TFAutoModelForVision2Seq", "TFAutoModelForVision2Seq",

View File

@@ -103,12 +103,12 @@ if is_tf_available():
"TFAutoModelForImageClassification", "TFAutoModelForImageClassification",
"TFAutoModelForMaskedLM", "TFAutoModelForMaskedLM",
"TFAutoModelForMultipleChoice", "TFAutoModelForMultipleChoice",
"TFAutoModelForNextSentencePrediction",
"TFAutoModelForPreTraining", "TFAutoModelForPreTraining",
"TFAutoModelForQuestionAnswering", "TFAutoModelForQuestionAnswering",
"TFAutoModelForSeq2SeqLM", "TFAutoModelForSeq2SeqLM",
"TFAutoModelForSequenceClassification", "TFAutoModelForSequenceClassification",
"TFAutoModelForSpeechSeq2Seq", "TFAutoModelForSpeechSeq2Seq",
"TFAutoModelForNextSentencePrediction",
"TFAutoModelForTableQuestionAnswering", "TFAutoModelForTableQuestionAnswering",
"TFAutoModelForTokenClassification", "TFAutoModelForTokenClassification",
"TFAutoModelForVision2Seq", "TFAutoModelForVision2Seq",

View File

@@ -335,6 +335,13 @@ class TFAutoModelForMultipleChoice(metaclass=DummyObject):
requires_backends(self, ["tf"]) requires_backends(self, ["tf"])
class TFAutoModelForNextSentencePrediction(metaclass=DummyObject):
_backends = ["tf"]
def __init__(self, *args, **kwargs):
requires_backends(self, ["tf"])
class TFAutoModelForPreTraining(metaclass=DummyObject): class TFAutoModelForPreTraining(metaclass=DummyObject):
_backends = ["tf"] _backends = ["tf"]