Skip to content

GitLab

Unverified Commit 10dd3a7d authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

[`SAM`] Change to `facebook/sam-vit-base` (#22891) 

change to `facebook/sam-vit-base`
parent aa43a765
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/transformers/models/sam/configuration_sam.py
View file @ 10dd3a7d
......@@ -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",
}
......
src/transformers/models/sam/modeling_sam.py
View file @ 10dd3a7d
......@@ -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
]
......
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