From 10dd3a7d1c0fff4485076ed6d488ccae0a792bd0 Mon Sep 17 00:00:00 2001 From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> Date: Thu, 20 Apr 2023 14:11:18 +0200 Subject: [PATCH] [`SAM`] Change to `facebook/sam-vit-base` (#22891) change to `facebook/sam-vit-base` --- src/transformers/models/sam/configuration_sam.py | 2 +- src/transformers/models/sam/modeling_sam.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/sam/configuration_sam.py b/src/transformers/models/sam/configuration_sam.py index 429362c9f6..fc3701fca7 100644 --- a/src/transformers/models/sam/configuration_sam.py +++ b/src/transformers/models/sam/configuration_sam.py @@ -25,7 +25,7 @@ logger = logging.get_logger(__name__) SAM_PRETRAINED_CONFIG_ARCHIVE_MAP = { "facebook/sam-vit-huge": "https://huggingface.co/facebook/sam-vit-huge/resolve/main/config.json", "facebook/sam-vit-large": "https://huggingface.co/facebook/sam-vit-large/resolve/main/config.json", - "facebook/sam-vit-big": "https://huggingface.co/facebook/sam-vit-big/resolve/main/config.json", + "facebook/sam-vit-base": "https://huggingface.co/facebook/sam-vit-base/resolve/main/config.json", } diff --git a/src/transformers/models/sam/modeling_sam.py b/src/transformers/models/sam/modeling_sam.py index d207f26468..f87cd77df6 100644 --- a/src/transformers/models/sam/modeling_sam.py +++ b/src/transformers/models/sam/modeling_sam.py @@ -40,7 +40,7 @@ _CHECKPOINT_FOR_DOC = "facebook/sam-vit-huge" SAM_PRETRAINED_MODEL_ARCHIVE_LIST = [ "facebook/sam-vit-huge", "facebook/sam-vit-large", - "facebook/sam-vit-big", + "facebook/sam-vit-base", # See all SAM models at https://huggingface.co/models?filter=sam ]