riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 18:51:14 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Update ESM checkpoints to point to facebook/ (#19675)

Browse Source 
* Update checkpoints to point to `facebook/`

* make fixup
This commit is contained in:
Matt 2022-10-17 18:09:24 +01:00 committed by GitHub
parent 4d77f18cba
commit 5fda1fbd46
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 10 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/transformers/models/esm/modeling_esm.py
View File

@ -42,13 +42,13 @@ from .configuration_esm import EsmConfig
logger = logging.get_logger(__name__) logger = logging.get_logger(__name__)
_CHECKPOINT_FOR_DOC = "Rocketknight1/esm2_t6_8M_UR50D" _CHECKPOINT_FOR_DOC = "facebook/esm2_t6_8M_UR50D"
_CONFIG_FOR_DOC = "EsmConfig" _CONFIG_FOR_DOC = "EsmConfig"
_TOKENIZER_FOR_DOC = "EsmTokenizer" _TOKENIZER_FOR_DOC = "EsmTokenizer"
ESM_PRETRAINED_MODEL_ARCHIVE_LIST = [ ESM_PRETRAINED_MODEL_ARCHIVE_LIST = [
"Rocketknight1/esm2_t6_8M_UR50D", "facebook/esm2_t6_8M_UR50D",
"Rocketknight1/esm2_t12_35M_UR50D", "facebook/esm2_t12_35M_UR50D",
# This is not a complete list of all ESM models! # This is not a complete list of all ESM models!
# See all ESM models at https://huggingface.co/models?filter=esm # See all ESM models at https://huggingface.co/models?filter=esm
] ]

6
src/transformers/models/esm/modeling_tf_esm.py
View File

@ -47,13 +47,13 @@ from .configuration_esm import EsmConfig
logger = logging.get_logger(__name__) logger = logging.get_logger(__name__)
_CHECKPOINT_FOR_DOC = "Rocketknight1/esm2_t6_8M_UR50D" _CHECKPOINT_FOR_DOC = "facebook/esm2_t6_8M_UR50D"
_CONFIG_FOR_DOC = "EsmConfig" _CONFIG_FOR_DOC = "EsmConfig"
_TOKENIZER_FOR_DOC = "EsmTokenizer" _TOKENIZER_FOR_DOC = "EsmTokenizer"
TF_ESM_PRETRAINED_MODEL_ARCHIVE_LIST = [ TF_ESM_PRETRAINED_MODEL_ARCHIVE_LIST = [
"Rocketknight1/esm2_t6_8M_UR50D", "facebook/esm2_t6_8M_UR50D",
"Rocketknight1/esm2_t12_35M_UR50D", "facebook/esm2_t12_35M_UR50D",
# This is not a complete list of all ESM models! # This is not a complete list of all ESM models!
# See all ESM models at https://huggingface.co/models?filter=esm # See all ESM models at https://huggingface.co/models?filter=esm
] ]

12
src/transformers/models/esm/tokenization_esm.py
View File

@ -27,18 +27,14 @@ VOCAB_FILES_NAMES = {"vocab_file": "vocab.txt"}
PRETRAINED_VOCAB_FILES_MAP = { PRETRAINED_VOCAB_FILES_MAP = {
"vocab_file": { "vocab_file": {
"Rocketknight1/esm2_t6_8M_UR50D": ( "facebook/esm2_t6_8M_UR50D": "https://huggingface.co/facebook/esm2_t6_8M_UR50D/resolve/main/vocab.txt",
"https://huggingface.co/Rocketknight1/esm2_t6_8M_UR50D/resolve/main/vocab.txt" "facebook/esm2_t12_35M_UR50D": "https://huggingface.co/facebook/esm2_t12_35M_UR50D/resolve/main/vocab.txt",
),
"Rocketknight1/esm2_t12_35M_UR50D": (
"https://huggingface.co/Rocketknight1/esm2_t12_35M_UR50D/resolve/main/vocab.txt"
),
}, },
} }
PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES = { PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES = {
"Rocketknight1/esm2_t6_8M_UR50D": 1024, "facebook/esm2_t6_8M_UR50D": 1024,
"Rocketknight1/esm2_t12_35M_UR50D": 1024, "facebook/esm2_t12_35M_UR50D": 1024,
} }