SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[DocTest] Fix correct checkpoint (#22988)

Browse Source 
fix pipeline issue
This commit is contained in:
Younes Belkada
2023-04-25 15:20:36 +02:00
committed by GitHub
parent 5427250351
commit a0ae2310ec
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/pipelines/mask_generation.py
View File

@@ -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",
... ) ... )