From 7259ecfc0ded6fffff617d3cabbc96f0dcc9e408 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Mon, 9 May 2022 11:14:36 -0400 Subject: [PATCH] Fix quality and repo consistency --- docs/source/en/serialization.mdx | 2 +- src/transformers/__init__.py | 2 +- src/transformers/models/auto/__init__.py | 2 +- src/transformers/utils/dummy_tf_objects.py | 7 +++++++ 4 files changed, 10 insertions(+), 3 deletions(-) diff --git a/docs/source/en/serialization.mdx b/docs/source/en/serialization.mdx index 510fbf936329..df39f1b97c54 100644 --- a/docs/source/en/serialization.mdx +++ b/docs/source/en/serialization.mdx @@ -68,7 +68,7 @@ Ready-made configurations include the following architectures: - M2M100 - Marian - mBART -- MobileBert +- MobileBERT - OpenAI GPT-2 - PLBart - RoBERTa diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py index 2c41ff883fb9..0888509290a9 100755 --- a/src/transformers/__init__.py +++ b/src/transformers/__init__.py @@ -1793,12 +1793,12 @@ "TFAutoModelForImageClassification", "TFAutoModelForMaskedLM", "TFAutoModelForMultipleChoice", + "TFAutoModelForNextSentencePrediction", "TFAutoModelForPreTraining", "TFAutoModelForQuestionAnswering", "TFAutoModelForSeq2SeqLM", "TFAutoModelForSequenceClassification", "TFAutoModelForSpeechSeq2Seq", - "TFAutoModelForNextSentencePrediction", "TFAutoModelForTableQuestionAnswering", "TFAutoModelForTokenClassification", "TFAutoModelForVision2Seq", diff --git a/src/transformers/models/auto/__init__.py b/src/transformers/models/auto/__init__.py index fa34a11964b0..644705a4eaf5 100644 --- a/src/transformers/models/auto/__init__.py +++ b/src/transformers/models/auto/__init__.py @@ -103,12 +103,12 @@ "TFAutoModelForImageClassification", "TFAutoModelForMaskedLM", "TFAutoModelForMultipleChoice", + "TFAutoModelForNextSentencePrediction", "TFAutoModelForPreTraining", "TFAutoModelForQuestionAnswering", "TFAutoModelForSeq2SeqLM", "TFAutoModelForSequenceClassification", "TFAutoModelForSpeechSeq2Seq", - "TFAutoModelForNextSentencePrediction", "TFAutoModelForTableQuestionAnswering", "TFAutoModelForTokenClassification", "TFAutoModelForVision2Seq", diff --git a/src/transformers/utils/dummy_tf_objects.py b/src/transformers/utils/dummy_tf_objects.py index e4a47290b5f5..e089a267a024 100644 --- a/src/transformers/utils/dummy_tf_objects.py +++ b/src/transformers/utils/dummy_tf_objects.py @@ -335,6 +335,13 @@ def __init__(self, *args, **kwargs): requires_backends(self, ["tf"]) +class TFAutoModelForNextSentencePrediction(metaclass=DummyObject): + _backends = ["tf"] + + def __init__(self, *args, **kwargs): + requires_backends(self, ["tf"]) + + class TFAutoModelForPreTraining(metaclass=DummyObject): _backends = ["tf"]