Spaces:

NeerjaK
/

DinoV3-Patch-Similarity-Visualiser

Sleeping

NeerjaK commited on Oct 22

Commit

cc0ae2a

1 Parent(s): 745d6a0

add token

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -8,12 +8,13 @@ import torch
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
-processor = AutoImageProcessor.from_pretrained("facebook/dinov3-vits16-pretrain-lvd1689m")
 model = AutoModel.from_pretrained(
     "facebook/dinov3-vits16-pretrain-lvd1689m",
     torch_dtype=torch.float16,
     device_map="auto",
-    attn_implementation="sdpa"
 )
 model.eval()

 device = 'cuda' if torch.cuda.is_available() else 'cpu'
+processor = AutoImageProcessor.from_pretrained("facebook/dinov3-vits16-pretrain-lvd1689m",use_auth_token=True)
 model = AutoModel.from_pretrained(
     "facebook/dinov3-vits16-pretrain-lvd1689m",
     torch_dtype=torch.float16,
     device_map="auto",
+    attn_implementation="sdpa",
+    use_auth_token=True
 )
 model.eval()