FIX: Hotfix HTTP error downloading model with hub (#552)

microsoft · Aug 4, 2022 · dbca94c · dbca94c
1 parent bdbbf3e
commit dbca94c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hi-ml-multimodal/test_multimodal/image/model/test_model.py b/hi-ml-multimodal/test_multimodal/image/model/test_model.py
@@ -143,7 +143,7 @@ def test_hubconf() -> None:
     """Test that instantiating the image model using the PyTorch Hub is consistent with older methods."""
     image = torch.rand(1, 3, 480, 480)
 
-    github = 'microsoft/hi-ml'
+    github = 'microsoft/hi-ml:main'
     model_hub = torch.hub.load(github, 'biovil_resnet', pretrained=True)
     model_himl = get_biovil_resnet()