facebook
/

metaclip-2-mt5-worldwide-b32

Zero-Shot Image Classification

Model card Files Files and versions

voidism commited on 26 days ago

Commit

fbbce52

·

verified ·

1 Parent(s): 6ef9a79

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -31,7 +31,7 @@ from transformers import pipeline
 clip = pipeline(
    task="zero-shot-image-classification",
-   model="voidism/metaclip-2-mt5-worldwide-b32",
    torch_dtype=torch.bfloat16,
    device=0
 )
@@ -50,8 +50,8 @@ from PIL import Image
 from transformers import AutoProcessor, AutoModel
 # note: make sure to verify that `AutoModel` is an instance of `MetaClip2Model`
-model = AutoModel.from_pretrained("voidism/metaclip-2-mt5-worldwide-b32", torch_dtype=torch.bfloat16, attn_implementation="sdpa")
-processor = AutoProcessor.from_pretrained("voidism/metaclip-2-mt5-worldwide-b32")
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)

 clip = pipeline(
    task="zero-shot-image-classification",
+   model="facebook/metaclip-2-mt5-worldwide-b32",
    torch_dtype=torch.bfloat16,
    device=0
 )
 from transformers import AutoProcessor, AutoModel
 # note: make sure to verify that `AutoModel` is an instance of `MetaClip2Model`
+model = AutoModel.from_pretrained("facebook/metaclip-2-mt5-worldwide-b32", torch_dtype=torch.bfloat16, attn_implementation="sdpa")
+processor = AutoProcessor.from_pretrained("facebook/metaclip-2-mt5-worldwide-b32")
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)