diff --git a/docs/source/en/main_classes/pipelines.md b/docs/source/en/main_classes/pipelines.md index 6a1175d1f6c..b105cb544ff 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 8001bb33c81..90147df41f2 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 0552988c7a4..42223e586fe 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",