From c34c50cdc01ed5a5dc73a2f36466d07e6fe9fb4e Mon Sep 17 00:00:00 2001
From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>
Date: Wed, 25 Oct 2023 19:31:36 +0200
Subject: [PATCH] [`docs`] Add `MaskGenerationPipeline` in docs (#27063)

* add `MaskGenerationPipeline` in docs

* Update __init__.py

* fix repo consistency and clarify docstring

* add on check docstirngs

* actually we do have a tf sam

* oops
---
 docs/source/en/main_classes/pipelines.md | 6 ++++++
 src/transformers/__init__.py             | 2 ++
 utils/check_docstrings.py                | 1 +
 3 files changed, 9 insertions(+)

diff --git a/docs/source/en/main_classes/pipelines.md b/docs/source/en/main_classes/pipelines.md
index 6a1175d1f6..b105cb544f 100644
--- a/docs/source/en/main_classes/pipelines.md
+++ b/docs/source/en/main_classes/pipelines.md
@@ -481,6 +481,12 @@ Pipelines available for multimodal tasks include the following.
     - __call__
     - all
 
+### MaskGenerationPipeline
+
+[[autodoc]] MaskGenerationPipeline
+    - __call__
+    - all
+
 ### VisualQuestionAnsweringPipeline
 
 [[autodoc]] VisualQuestionAnsweringPipeline
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
index 8001bb33c8..90147df41f 100644
--- a/src/transformers/__init__.py
+++ b/src/transformers/__init__.py
@@ -672,6 +672,7 @@ _import_structure = {
         "ImageToImagePipeline",
         "ImageToTextPipeline",
         "JsonPipelineDataFormat",
+        "MaskGenerationPipeline",
         "NerPipeline",
         "ObjectDetectionPipeline",
         "PipedPipelineDataFormat",
@@ -4814,6 +4815,7 @@ if TYPE_CHECKING:
         ImageToImagePipeline,
         ImageToTextPipeline,
         JsonPipelineDataFormat,
+        MaskGenerationPipeline,
         NerPipeline,
         ObjectDetectionPipeline,
         PipedPipelineDataFormat,
diff --git a/utils/check_docstrings.py b/utils/check_docstrings.py
index 0552988c7a..42223e586f 100644
--- a/utils/check_docstrings.py
+++ b/utils/check_docstrings.py
@@ -356,6 +356,7 @@ OBJECTS_TO_IGNORE = [
     "M2M100Config",
     "M2M100Tokenizer",
     "MarkupLMProcessor",
+    "MaskGenerationPipeline",
     "MBart50TokenizerFast",
     "MBartConfig",
     "MCTCTFeatureExtractor",