Unverified Commit 3a396c59 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

fix: ckpt paths. (#19159)

parent 74a3ea47
......@@ -36,9 +36,9 @@ logger = logging.get_logger(__name__)
_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
]
......
......@@ -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()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment