[docs] Add MaskGenerationPipeline in docs (huggingface#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
ArthurZucker · Oct 25, 2023 · c34c50c · c34c50c
1 parent ba073ea
commit c34c50c
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git 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
@@ -672,6 +672,7 @@
  "ImageToImagePipeline",
  "ImageToTextPipeline",
  "JsonPipelineDataFormat",
+ "MaskGenerationPipeline",
  "NerPipeline",
  "ObjectDetectionPipeline",
  "PipedPipelineDataFormat",
@@ -4814,6 +4815,7 @@
  ImageToImagePipeline,
  ImageToTextPipeline,
  JsonPipelineDataFormat,
+ MaskGenerationPipeline,
  NerPipeline,
  ObjectDetectionPipeline,
  PipedPipelineDataFormat,

diff --git a/utils/check_docstrings.py b/utils/check_docstrings.py
@@ -356,6 +356,7 @@
  "M2M100Config",
  "M2M100Tokenizer",
  "MarkupLMProcessor",
+ "MaskGenerationPipeline",
  "MBart50TokenizerFast",
  "MBartConfig",
  "MCTCTFeatureExtractor",