panda1835 commited on
Commit
abc71eb
·
1 Parent(s): 206c7b2

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +2 -0
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(