fix: ckpt paths. (huggingface#19159)

oneraghavan · Sep 22, 2022 · 3a396c5 · 3a396c5
1 parent 74a3ea4
commit 3a396c5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/transformers/models/vit_msn/modeling_vit_msn.py b/src/transformers/models/vit_msn/modeling_vit_msn.py
@@ -36,9 +36,9 @@
 
 
 _CONFIG_FOR_DOC = "ViTMSNConfig"
-_CHECKPOINT_FOR_DOC = "sayakpaul/vit-msn-small"
+_CHECKPOINT_FOR_DOC = "facebook/vit-msn-small"
 VIT_MSN_PRETRAINED_MODEL_ARCHIVE_LIST = [
-    "sayakpaul/vit-msn-small",
+    "facebook/vit-msn-small",
     # See all ViTMSN models at https://huggingface.co/models?filter=vit_msn
 ]
 

diff --git a/tests/models/vit_msn/test_modeling_vit_msn.py b/tests/models/vit_msn/test_modeling_vit_msn.py
@@ -215,12 +215,12 @@ def prepare_img():
 class ViTMSNModelIntegrationTest(unittest.TestCase):
     @cached_property
     def default_feature_extractor(self):
-        return ViTFeatureExtractor.from_pretrained("sayakpaul/vit-msn-small") if is_vision_available() else None
+        return ViTFeatureExtractor.from_pretrained("facebook/vit-msn-small") if is_vision_available() else None
 
     @slow
     def test_inference_image_classification_head(self):
         torch.manual_seed(2)
-        model = ViTMSNForImageClassification.from_pretrained("sayakpaul/vit-msn-small").to(torch_device)
+        model = ViTMSNForImageClassification.from_pretrained("facebook/vit-msn-small").to(torch_device)
 
         feature_extractor = self.default_feature_extractor
         image = prepare_img()