From 2d280c6beaab504952479a0f622889c7b3eefc91 Mon Sep 17 00:00:00 2001 From: younesbelkada Date: Tue, 25 Apr 2023 12:46:22 +0000 Subject: [PATCH] fix pipeline issue --- src/transformers/pipelines/mask_generation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/pipelines/mask_generation.py b/src/transformers/pipelines/mask_generation.py index 650d4a41416ff9..bfcd9e800a87e8 100644 --- 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", ... )