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__) ...@@ -36,9 +36,9 @@ logger = logging.get_logger(__name__)
_CONFIG_FOR_DOC = "ViTMSNConfig" _CONFIG_FOR_DOC = "ViTMSNConfig"
_CHECKPOINT_FOR_DOC = "sayakpaul/vit-msn-small" _CHECKPOINT_FOR_DOC = "facebook/vit-msn-small"
VIT_MSN_PRETRAINED_MODEL_ARCHIVE_LIST = [ 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 # See all ViTMSN models at https://huggingface.co/models?filter=vit_msn
] ]
......
...@@ -215,12 +215,12 @@ def prepare_img(): ...@@ -215,12 +215,12 @@ def prepare_img():
class ViTMSNModelIntegrationTest(unittest.TestCase): class ViTMSNModelIntegrationTest(unittest.TestCase):
@cached_property @cached_property
def default_feature_extractor(self): 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 @slow
def test_inference_image_classification_head(self): def test_inference_image_classification_head(self):
torch.manual_seed(2) 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 feature_extractor = self.default_feature_extractor
image = prepare_img() 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