Update I-JEPA checkpoints path (#35120)

Update checkpoints path
This commit is contained in:
Pavel Iakubovskii
2024-12-06 13:42:51 +00:00
committed by GitHub
parent 7f95372c62
commit c8c8dffbe4
3 changed files with 7 additions and 7 deletions

View File

@@ -45,7 +45,7 @@ url_2 = "http://images.cocodataset.org/val2017/000000219578.jpg"
image_1 = Image.open(requests.get(url_1, stream=True).raw)
image_2 = Image.open(requests.get(url_2, stream=True).raw)
model_id = "jmtzt/ijepa_vith14_1k"
model_id = "facebook/ijepa_vith14_1k"
processor = AutoProcessor.from_pretrained(model_id)
model = AutoModel.from_pretrained(model_id)