From f1e42bc50e47c66360fb9d2edf55c742ce6a8245 Mon Sep 17 00:00:00 2001
From: Yih-Dar <2521628+ydshieh@users.noreply.github.com>
Date: Thu, 27 Oct 2022 14:38:59 +0200
Subject: [PATCH] Some fixes regarding auto mappings and test class names
 (#19923)

* Add pegasus_x

* ViTMSN

* ESM

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
---
 src/transformers/models/auto/tokenization_auto.py | 7 +++++++
 src/transformers/models/esm/configuration_esm.py  | 6 +++---
 tests/models/vit_msn/test_modeling_vit_msn.py     | 8 ++++----
 3 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/src/transformers/models/auto/tokenization_auto.py b/src/transformers/models/auto/tokenization_auto.py
index 43aaa0115..46e57ac58 100644
--- a/src/transformers/models/auto/tokenization_auto.py
+++ b/src/transformers/models/auto/tokenization_auto.py
@@ -205,6 +205,13 @@ else:
                     "PegasusTokenizerFast" if is_tokenizers_available() else None,
                 ),
             ),
+            (
+                "pegasus_x",
+                (
+                    "PegasusTokenizer" if is_sentencepiece_available() else None,
+                    "PegasusTokenizerFast" if is_tokenizers_available() else None,
+                ),
+            ),
             (
                 "perceiver",
                 (
diff --git a/src/transformers/models/esm/configuration_esm.py b/src/transformers/models/esm/configuration_esm.py
index 6b8f241ff..8f184c3e0 100644
--- a/src/transformers/models/esm/configuration_esm.py
+++ b/src/transformers/models/esm/configuration_esm.py
@@ -21,7 +21,7 @@ from ...utils import logging
 logger = logging.get_logger(__name__)
 
 ESM_PRETRAINED_CONFIG_ARCHIVE_MAP = {
-    "facebook/esm1b": "https://huggingface.co/facebook/esm1b/resolve/main/config.json",
+    "facebook/esm-1b": "https://huggingface.co/facebook/esm-1b/resolve/main/config.json",
     # See all ESM models at https://huggingface.co/models?filter=esm
 }
 
@@ -31,7 +31,7 @@ class EsmConfig(PretrainedConfig):
     This is the configuration class to store the configuration of a [`ESMModel`]. It is used to instantiate a ESM model
     according to the specified arguments, defining the model architecture. Instantiating a configuration with the
     defaults will yield a similar configuration to that of the ESM
-    [esm-base-uncased](https://huggingface.co/esm-base-uncased) architecture.
+    [facebook/esm-1b](https://huggingface.co/facebook/esm-1b) architecture.
 
     Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the
     documentation from [`PretrainedConfig`] for more information.
@@ -90,7 +90,7 @@ class EsmConfig(PretrainedConfig):
     ```python
     >>> from transformers import EsmModel, EsmConfig
 
-    >>> # Initializing a ESM esm-base-uncased style configuration >>> configuration = EsmConfig()
+    >>> # Initializing a ESM facebook/esm-1b style configuration >>> configuration = EsmConfig()
 
     >>> # Initializing a model from the configuration >>> model = ESMModel(configuration)
 
diff --git a/tests/models/vit_msn/test_modeling_vit_msn.py b/tests/models/vit_msn/test_modeling_vit_msn.py
index 0cba3dad0..b93e5c72d 100644
--- a/tests/models/vit_msn/test_modeling_vit_msn.py
+++ b/tests/models/vit_msn/test_modeling_vit_msn.py
@@ -40,7 +40,7 @@ if is_vision_available():
     from transformers import ViTFeatureExtractor
 
 
-class ViTMAEModelTester:
+class ViTMSNModelTester:
     def __init__(
         self,
         parent,
@@ -146,7 +146,7 @@ class ViTMAEModelTester:
 @require_torch
 class ViTMSNModelTest(ModelTesterMixin, unittest.TestCase):
     """
-    Here we also overwrite some of the tests of test_modeling_common.py, as ViTMAE does not use input_ids, inputs_embeds,
+    Here we also overwrite some of the tests of test_modeling_common.py, as ViTMSN does not use input_ids, inputs_embeds,
     attention_mask and seq_length.
     """
 
@@ -158,13 +158,13 @@ class ViTMSNModelTest(ModelTesterMixin, unittest.TestCase):
     test_head_masking = False
 
     def setUp(self):
-        self.model_tester = ViTMAEModelTester(self)
+        self.model_tester = ViTMSNModelTester(self)
         self.config_tester = ConfigTester(self, config_class=ViTMSNConfig, has_text_modality=False, hidden_size=37)
 
     def test_config(self):
         self.config_tester.run_common_tests()
 
-    @unittest.skip(reason="ViTMAE does not use inputs_embeds")
+    @unittest.skip(reason="ViTMSN does not use inputs_embeds")
     def test_inputs_embeds(self):
         pass