Skip to content

GitLab

Unverified Commit d3d5fa3e authored by Matt's avatar Matt Committed by GitHub
Browse files

Add missing ESM autoclass (#20177) 

* Add missing ESM autoclass

* Correct ESMFold checkpoint
parent 92cfe8b0
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
src/transformers/models/auto/modeling_auto.py
View file @ d3d5fa3e
...@@ -445,6 +445,7 @@ MODEL_FOR_MASKED_LM_MAPPING_NAMES = OrderedDict( ...@@ -445,6 +445,7 @@ MODEL_FOR_MASKED_LM_MAPPING_NAMES = OrderedDict(
("distilbert", "DistilBertForMaskedLM"), ("distilbert", "DistilBertForMaskedLM"),
("electra", "ElectraForMaskedLM"), ("electra", "ElectraForMaskedLM"),
("ernie", "ErnieForMaskedLM"), ("ernie", "ErnieForMaskedLM"),
("esm", "EsmForMaskedLM"),
("flaubert", "FlaubertWithLMHeadModel"), ("flaubert", "FlaubertWithLMHeadModel"),
("fnet", "FNetForMaskedLM"), ("fnet", "FNetForMaskedLM"),
("funnel", "FunnelForMaskedLM"), ("funnel", "FunnelForMaskedLM"),
......
src/transformers/models/esm/modeling_esmfold.py
View file @ d3d5fa3e
...@@ -52,7 +52,7 @@ from .openfold_utils import ( ...@@ -52,7 +52,7 @@ from .openfold_utils import (
logger = logging.get_logger(__name__) logger = logging.get_logger(__name__)
_CHECKPOINT_FOR_DOC = "Rocketknight1/esmfold_v1" _CHECKPOINT_FOR_DOC = "facebook/esmfold_v1"
_CONFIG_FOR_DOC = "EsmConfig" _CONFIG_FOR_DOC = "EsmConfig"
_TOKENIZER_FOR_DOC = "EsmTokenizer" _TOKENIZER_FOR_DOC = "EsmTokenizer"
......
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