[DocTest] Fix correct checkpoint (huggingface#22988)

younesbelkada · web-flow · commit a0ae2310ec46 · 2023-04-25T15:20:36.000+02:00
fix pipeline issue
diff --git a/src/transformers/pipelines/mask_generation.py b/src/transformers/pipelines/mask_generation.py
@@ -70,7 +70,7 @@ class MaskGenerationPipeline(ChunkPipeline):
     ```python
     >>> from transformers import pipeline
 
-    >>> generator = pipeline(model="facebook/sam-vit-h", task="mask-generation")
+    >>> generator = pipeline(model="facebook/sam-vit-base", task="mask-generation")
     >>> outputs = generator(
     ...     "http://images.cocodataset.org/val2017/000000039769.jpg",
     ... )
