From 3a396c59b8187cb92c6ea56da6d4c69b4e27cf62 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Thu, 22 Sep 2022 20:33:01 +0530 Subject: [PATCH] fix: ckpt paths. (#19159) --- src/transformers/models/vit_msn/modeling_vit_msn.py | 4 ++-- tests/models/vit_msn/test_modeling_vit_msn.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/transformers/models/vit_msn/modeling_vit_msn.py b/src/transformers/models/vit_msn/modeling_vit_msn.py index 314b3dbd5bfdbd..a190c42caa707b 100644 --- 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 index b858da42b3d4e3..0cba3dad009429 100644 --- 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()