Instructions to use facebook/sam-vit-huge with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use facebook/sam-vit-huge with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("mask-generation", model="facebook/sam-vit-huge")# Load model directly from transformers import AutoProcessor, AutoModelForMaskGeneration processor = AutoProcessor.from_pretrained("facebook/sam-vit-huge") model = AutoModelForMaskGeneration.from_pretrained("facebook/sam-vit-huge") - Notebooks
- Google Colab
- Kaggle
Usage code needs reviewing
#4
by mvetter - opened
There seem to be some mistakes in the Usage section of the model card.
In the section "prompted-mask-generation": there is a typo: "processsor" (3 's')
In the section "prompted-mask-generation": use of .to(device) without setting device
In the section "Automatic-Mask-Generation": running the provided code yields "KeyError: "Unknown task automatic-mask-generation"" when running transformers 4.29.0.dev0 installed from github