Fix quality and repo consistency

elusenji · Jun 12, 2022 · 7259ecf · 7259ecf
1 parent b319e1e
commit 7259ecf
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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"]
-Original file line number
+Diff line change
@@ Expand Up @@
     - M2M100
     - Marian
     - mBART
-    - MobileBert
+    - MobileBERT
     - OpenAI GPT-2
     - PLBart
     - RoBERTa
@@ Expand Down @@