From 3d2d58c91a67a09fe5a754871c6110824fe3652a Mon Sep 17 00:00:00 2001 From: ydshieh Date: Fri, 12 Aug 2022 13:40:29 +0200 Subject: [PATCH 1/3] fix init --- src/transformers/__init__.py | 2 ++ src/transformers/models/auto/__init__.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py index be2be2727f01..2f53db07f078 100755 --- a/src/transformers/__init__.py +++ b/src/transformers/__init__.py @@ -2105,6 +2105,7 @@ "TFAutoModelForNextSentencePrediction", "TFAutoModelForPreTraining", "TFAutoModelForQuestionAnswering", + "TFAutoModelForSemanticSegmentation", "TFAutoModelForSeq2SeqLM", "TFAutoModelForSequenceClassification", "TFAutoModelForSpeechSeq2Seq", @@ -4599,6 +4600,7 @@ TFAutoModelForNextSentencePrediction, TFAutoModelForPreTraining, TFAutoModelForQuestionAnswering, + TFAutoModelForSemanticSegmentation, TFAutoModelForSeq2SeqLM, TFAutoModelForSequenceClassification, TFAutoModelForSpeechSeq2Seq, diff --git a/src/transformers/models/auto/__init__.py b/src/transformers/models/auto/__init__.py index 139d4feda336..ec253f6037a3 100644 --- a/src/transformers/models/auto/__init__.py +++ b/src/transformers/models/auto/__init__.py @@ -128,6 +128,7 @@ "TFAutoModelForNextSentencePrediction", "TFAutoModelForPreTraining", "TFAutoModelForQuestionAnswering", + "TFAutoModelForSemanticSegmentation", "TFAutoModelForSeq2SeqLM", "TFAutoModelForSequenceClassification", "TFAutoModelForSpeechSeq2Seq", @@ -271,6 +272,7 @@ TFAutoModelForNextSentencePrediction, TFAutoModelForPreTraining, TFAutoModelForQuestionAnswering, + TFAutoModelForSemanticSegmentation, TFAutoModelForSeq2SeqLM, TFAutoModelForSequenceClassification, TFAutoModelForSpeechSeq2Seq, From 07a7efff65fe9cb7fbb7524627612d215000d922 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Fri, 12 Aug 2022 13:44:55 +0200 Subject: [PATCH 2/3] fix init --- src/transformers/utils/dummy_tf_objects.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/transformers/utils/dummy_tf_objects.py b/src/transformers/utils/dummy_tf_objects.py index 6df601ca646a..fec5ffe70080 100644 --- a/src/transformers/utils/dummy_tf_objects.py +++ b/src/transformers/utils/dummy_tf_objects.py @@ -362,6 +362,13 @@ def __init__(self, *args, **kwargs): 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"] From 3bdc76f09b94471cfaba22535ad4e23e1ebbe1d8 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Fri, 12 Aug 2022 13:53:11 +0200 Subject: [PATCH 3/3] fix --- docs/source/en/model_doc/auto.mdx | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/docs/source/en/model_doc/auto.mdx b/docs/source/en/model_doc/auto.mdx index 67fc81d280a7..995296485b94 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