Unverified Commit a0ae2310 authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

[`DocTest`] Fix correct checkpoint (#22988)

fix pipeline issue
parent 54272503
...@@ -70,7 +70,7 @@ class MaskGenerationPipeline(ChunkPipeline): ...@@ -70,7 +70,7 @@ class MaskGenerationPipeline(ChunkPipeline):
```python ```python
>>> from transformers import pipeline >>> 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( >>> outputs = generator(
... "http://images.cocodataset.org/val2017/000000039769.jpg", ... "http://images.cocodataset.org/val2017/000000039769.jpg",
... ) ... )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment