SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source 
change to `facebook/sam-vit-base`
This commit is contained in:
Younes Belkada
2023-04-20 14:11:18 +02:00
committed by GitHub
parent aa43a76538
commit 10dd3a7d1c
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/sam/configuration_sam.py
View File

@@ -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",
}

2
src/transformers/models/sam/modeling_sam.py
View File

@@ -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
]