From 2156619f10b33fb58520830bff0a92c36d13128e Mon Sep 17 00:00:00 2001
From: Yih-Dar <2521628+ydshieh@users.noreply.github.com>
Date: Fri, 12 Aug 2022 15:10:00 +0200
Subject: [PATCH] Add `TFAutoModelForSemanticSegmentation` to the main
 `__init__.py` (#18600)

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
---
 docs/source/en/model_doc/auto.mdx          | 4 ++++
 src/transformers/__init__.py               | 2 ++
 src/transformers/models/auto/__init__.py   | 2 ++
 src/transformers/utils/dummy_tf_objects.py | 7 +++++++
 4 files changed, 15 insertions(+)

diff --git a/docs/source/en/model_doc/auto.mdx b/docs/source/en/model_doc/auto.mdx
index 67fc81d280..995296485b 100644
--- a/docs/source/en/model_doc/auto.mdx
+++ b/docs/source/en/model_doc/auto.mdx
@@ -186,6 +186,10 @@ Likewise, if your `NewModel` is a subclass of [`PreTrainedModel`], make sure its
 
 [[autodoc]] TFAutoModelForImageClassification
 
+## TFAutoModelForSemanticSegmentation
+
+[[autodoc]] TFAutoModelForSemanticSegmentation
+
 ## TFAutoModelForMaskedLM
 
 [[autodoc]] TFAutoModelForMaskedLM
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
index be2be2727f..2f53db07f0 100755
--- a/src/transformers/__init__.py
+++ b/src/transformers/__init__.py
@@ -2105,6 +2105,7 @@ else:
             "TFAutoModelForNextSentencePrediction",
             "TFAutoModelForPreTraining",
             "TFAutoModelForQuestionAnswering",
+            "TFAutoModelForSemanticSegmentation",
             "TFAutoModelForSeq2SeqLM",
             "TFAutoModelForSequenceClassification",
             "TFAutoModelForSpeechSeq2Seq",
@@ -4599,6 +4600,7 @@ if TYPE_CHECKING:
             TFAutoModelForNextSentencePrediction,
             TFAutoModelForPreTraining,
             TFAutoModelForQuestionAnswering,
+            TFAutoModelForSemanticSegmentation,
             TFAutoModelForSeq2SeqLM,
             TFAutoModelForSequenceClassification,
             TFAutoModelForSpeechSeq2Seq,
diff --git a/src/transformers/models/auto/__init__.py b/src/transformers/models/auto/__init__.py
index 139d4feda3..ec253f6037 100644
--- a/src/transformers/models/auto/__init__.py
+++ b/src/transformers/models/auto/__init__.py
@@ -128,6 +128,7 @@ else:
         "TFAutoModelForNextSentencePrediction",
         "TFAutoModelForPreTraining",
         "TFAutoModelForQuestionAnswering",
+        "TFAutoModelForSemanticSegmentation",
         "TFAutoModelForSeq2SeqLM",
         "TFAutoModelForSequenceClassification",
         "TFAutoModelForSpeechSeq2Seq",
@@ -271,6 +272,7 @@ if TYPE_CHECKING:
             TFAutoModelForNextSentencePrediction,
             TFAutoModelForPreTraining,
             TFAutoModelForQuestionAnswering,
+            TFAutoModelForSemanticSegmentation,
             TFAutoModelForSeq2SeqLM,
             TFAutoModelForSequenceClassification,
             TFAutoModelForSpeechSeq2Seq,
diff --git a/src/transformers/utils/dummy_tf_objects.py b/src/transformers/utils/dummy_tf_objects.py
index 6df601ca64..fec5ffe700 100644
--- a/src/transformers/utils/dummy_tf_objects.py
+++ b/src/transformers/utils/dummy_tf_objects.py
@@ -362,6 +362,13 @@ class TFAutoModelForQuestionAnswering(metaclass=DummyObject):
         requires_backends(self, ["tf"])
 
 
+class TFAutoModelForSemanticSegmentation(metaclass=DummyObject):
+    _backends = ["tf"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["tf"])
+
+
 class TFAutoModelForSeq2SeqLM(metaclass=DummyObject):
     _backends = ["tf"]