Spaces:
Running
Running
Update models.py
Browse files
models.py
CHANGED
@@ -10,6 +10,8 @@ class DinoVisionTransformerClassifier(nn.Module):
|
|
10 |
|
11 |
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
12 |
|
|
|
|
|
13 |
self.model = torch.hub.load("facebookresearch/dinov2", "dinov2_vitb14_lc")
|
14 |
|
15 |
self.model.linear_head = nn.Sequential(
|
|
|
10 |
|
11 |
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
12 |
|
13 |
+
# Workaround to bypass HTTP Error 403 rate limit exceeded
|
14 |
+
torch.hub._validate_not_a_forked_repo=lambda a,b,c: True
|
15 |
self.model = torch.hub.load("facebookresearch/dinov2", "dinov2_vitb14_lc")
|
16 |
|
17 |
self.model.linear_head = nn.Sequential(
|