Fix checkpoint for ViT Config

f4b7420d · Sylvain Gugger · 6a3c883c · f4b7420d
Commit f4b7420d authored Jan 19, 2022 by Sylvain Gugger
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/vit/configuration_vit.py src/transformers/models/vit/configuration_vit.py +1 -1

No files found.
--- a/src/transformers/models/vit/configuration_vit.py
+++ b/src/transformers/models/vit/configuration_vit.py
@@ -21,7 +21,7 @@ from ...utils import logging
 logger = logging.get_logger(__name__)

 VIT_PRETRAINED_CONFIG_ARCHIVE_MAP = {
-    "nielsr/vit-base-patch16-224": "https://huggingface.co/vit-base-patch16-224/resolve/main/config.json",
+    "google/vit-base-patch16-224": "https://huggingface.co/vit-base-patch16-224/resolve/main/config.json",
    # See all ViT models at https://huggingface.co/models?filter=vit
 }