From 3d852da2db3b1314be466f1ef7f65270611d2914 Mon Sep 17 00:00:00 2001
From: fxmarty <9808326+fxmarty@users.noreply.github.com>
Date: Fri, 21 Apr 2023 16:04:45 +0200
Subject: [PATCH] Expose AutoModelForMaskGeneration (#22910)

* expose

* style

* add dummy object

* amazed by the quality of transformers CI
---
 docs/source/en/model_doc/auto.mdx          | 4 ++++
 src/transformers/__init__.py               | 2 ++
 src/transformers/models/auto/__init__.py   | 2 ++
 src/transformers/utils/dummy_pt_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 39b0645eb5..0f82f383da 100644
--- a/docs/source/en/model_doc/auto.mdx
+++ b/docs/source/en/model_doc/auto.mdx
@@ -134,6 +134,10 @@ The following auto classes are available for the following natural language proc
 
 [[autodoc]] FlaxAutoModelForMaskedLM
 
+### AutoModelForMaskGeneration
+
+[[autodoc]] AutoModelForMaskGeneration
+
 ### AutoModelForSeq2SeqLM
 
 [[autodoc]] AutoModelForSeq2SeqLM
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
index 6171da7190..3fd5b3deac 100644
--- a/src/transformers/__init__.py
+++ b/src/transformers/__init__.py
@@ -1046,6 +1046,7 @@ else:
             "AutoModelForInstanceSegmentation",
             "AutoModelForMaskedImageModeling",
             "AutoModelForMaskedLM",
+            "AutoModelForMaskGeneration",
             "AutoModelForMultipleChoice",
             "AutoModelForNextSentencePrediction",
             "AutoModelForObjectDetection",
@@ -4687,6 +4688,7 @@ if TYPE_CHECKING:
             AutoModelForInstanceSegmentation,
             AutoModelForMaskedImageModeling,
             AutoModelForMaskedLM,
+            AutoModelForMaskGeneration,
             AutoModelForMultipleChoice,
             AutoModelForNextSentencePrediction,
             AutoModelForObjectDetection,
diff --git a/src/transformers/models/auto/__init__.py b/src/transformers/models/auto/__init__.py
index 944b0ccb38..7ea870a933 100644
--- a/src/transformers/models/auto/__init__.py
+++ b/src/transformers/models/auto/__init__.py
@@ -83,6 +83,7 @@ else:
         "AutoModelForImageClassification",
         "AutoModelForImageSegmentation",
         "AutoModelForInstanceSegmentation",
+        "AutoModelForMaskGeneration",
         "AutoModelForMaskedImageModeling",
         "AutoModelForMaskedLM",
         "AutoModelForMultipleChoice",
@@ -250,6 +251,7 @@ if TYPE_CHECKING:
             AutoModelForInstanceSegmentation,
             AutoModelForMaskedImageModeling,
             AutoModelForMaskedLM,
+            AutoModelForMaskGeneration,
             AutoModelForMultipleChoice,
             AutoModelForNextSentencePrediction,
             AutoModelForObjectDetection,
diff --git a/src/transformers/utils/dummy_pt_objects.py b/src/transformers/utils/dummy_pt_objects.py
index 9b607e918e..5409d8a27b 100644
--- a/src/transformers/utils/dummy_pt_objects.py
+++ b/src/transformers/utils/dummy_pt_objects.py
@@ -639,6 +639,13 @@ class AutoModelForMaskedLM(metaclass=DummyObject):
         requires_backends(self, ["torch"])
 
 
+class AutoModelForMaskGeneration(metaclass=DummyObject):
+    _backends = ["torch"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["torch"])
+
+
 class AutoModelForMultipleChoice(metaclass=DummyObject):
     _backends = ["torch"]