From a0ae2310ec46a2c592950babc85cf02e325bf6a7 Mon Sep 17 00:00:00 2001 From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> Date: Tue, 25 Apr 2023 15:20:36 +0200 Subject: [PATCH] [`DocTest`] Fix correct checkpoint (#22988) 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 650d4a4141..bfcd9e800a 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", ... )