Spaces:
Sleeping
Sleeping
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(
|