I’m trying to run the code in this tutorial made by HF to run the new Segment Anything model.
Problem is, I can’t actually import it.
Here is the code:
from transformers import pipeline
generator = pipeline("mask-generation", model="facebook/sam-vit-huge", device=0)
And the error:
641
642 def __getitem__(self, key):
--> 643 if key in self._extra_content:
644 return self._extra_content[key]
645 if key not in self._mapping:
KeyError: 'sam'
Am I missing something here?