From f414e108b6ffceb314931c08c6a5c618c380167b Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Wed, 20 Apr 2022 11:59:47 +0200 Subject: [PATCH] Add missing YOLOS_PRETRAINED_CONFIG_ARCHIVE_MAP --- src/transformers/models/yolos/configuration_yolos.py | 6 +++--- src/transformers/models/yolos/modeling_yolos.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/transformers/models/yolos/configuration_yolos.py b/src/transformers/models/yolos/configuration_yolos.py index 5010a459926f88..24dc8edb0eedb4 100644 --- a/src/transformers/models/yolos/configuration_yolos.py +++ b/src/transformers/models/yolos/configuration_yolos.py @@ -20,9 +20,9 @@ logger = logging.get_logger(__name__) -VIT_PRETRAINED_CONFIG_ARCHIVE_MAP = { - "google/vit-base-patch16-224": "https://huggingface.co/vit-base-patch16-224/resolve/main/config.json", - # See all YOLOS models at https://huggingface.co/models?filter=vit +YOLOS_PRETRAINED_CONFIG_ARCHIVE_MAP = { + "google/yolos-s": "https://huggingface.co/yolos-s/resolve/main/config.json", + # See all YOLOS models at https://huggingface.co/models?filter=yolos } diff --git a/src/transformers/models/yolos/modeling_yolos.py b/src/transformers/models/yolos/modeling_yolos.py index 7adf7b80ca5c11..2390b5cad88036 100755 --- a/src/transformers/models/yolos/modeling_yolos.py +++ b/src/transformers/models/yolos/modeling_yolos.py @@ -544,7 +544,7 @@ class YolosPreTrainedModel(PreTrainedModel): """ config_class = YolosConfig - base_model_prefix = "yolos" + base_model_prefix = "vit" main_input_name = "pixel_values" supports_gradient_checkpointing = True