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

* Update checkpoints to point to `facebook/`

* make fixup

Author: Rocketknight1

src/transformers/models/esm/modeling_esm.py

@@ -42,13 +42,13 @@
 
 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"
 _TOKENIZER_FOR_DOC = "EsmTokenizer"
 
 ESM_PRETRAINED_MODEL_ARCHIVE_LIST = [
-    "Rocketknight1/esm2_t6_8M_UR50D",
-    "Rocketknight1/esm2_t12_35M_UR50D",
+    "facebook/esm2_t6_8M_UR50D",
+    "facebook/esm2_t12_35M_UR50D",
     # This is not a complete list of all ESM models!
     # See all ESM models at https://huggingface.co/models?filter=esm
 ]

src/transformers/models/esm/modeling_tf_esm.py

@@ -47,13 +47,13 @@
 
 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"
 _TOKENIZER_FOR_DOC = "EsmTokenizer"
 
 TF_ESM_PRETRAINED_MODEL_ARCHIVE_LIST = [
-    "Rocketknight1/esm2_t6_8M_UR50D",
-    "Rocketknight1/esm2_t12_35M_UR50D",
+    "facebook/esm2_t6_8M_UR50D",
+    "facebook/esm2_t12_35M_UR50D",
     # This is not a complete list of all ESM models!
     # See all ESM models at https://huggingface.co/models?filter=esm
 ]

src/transformers/models/esm/tokenization_esm.py

@@ -27,18 +27,14 @@
 
 PRETRAINED_VOCAB_FILES_MAP = {
     "vocab_file": {
-        "Rocketknight1/esm2_t6_8M_UR50D": (
-            "https://huggingface.co/Rocketknight1/esm2_t6_8M_UR50D/resolve/main/vocab.txt"
-        ),
-        "Rocketknight1/esm2_t12_35M_UR50D": (
-            "https://huggingface.co/Rocketknight1/esm2_t12_35M_UR50D/resolve/main/vocab.txt"
-        ),
+        "facebook/esm2_t6_8M_UR50D": "https://huggingface.co/facebook/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",
     },
 }
 
 PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES = {
-    "Rocketknight1/esm2_t6_8M_UR50D": 1024,
-    "Rocketknight1/esm2_t12_35M_UR50D": 1024,
+    "facebook/esm2_t6_8M_UR50D": 1024,
+    "facebook/esm2_t12_35M_UR50D": 1024,
 }