Merge pull request #177 from biigle/patch-1

Specfiy branch name for torch.hub model
biigle · Jul 11, 2024 · 62464ee · 62464ee
2 parents d25d263 + c45cfd3
commit 62464ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/resources/scripts/ExtractFeatures.py b/src/resources/scripts/ExtractFeatures.py
@@ -19,7 +19,7 @@
 with open(sys.argv[1], 'r') as f:
     input_json = json.load(f)
 
-dinov2_vits14 = hub_load('facebookresearch/dinov2', 'dinov2_vits14')
+dinov2_vits14 = hub_load('facebookresearch/dinov2:main', 'dinov2_vits14')
 
 if cuda_is_available():
     device = device('cuda')