Add warning message for beta and gamma parameters

src/transformers/modeling_utils.py

@@ -663,7 +663,15 @@ def _load_state_dict_into_model(model_to_load, state_dict, start_prefix):
         new_key = None
         if "gamma" in key:
             new_key = key.replace("gamma", "weight")
+            logger.warning(
+                "A parameter name that contains `gamma` will be renamed internally. Please use a different name "
+                "to suppress this warning."
+            )
         if "beta" in key:
+            logger.warning(
+                "A parameter name that contains `beta` will be renamed internally. Please use a different name "
+                "to suppress this warning."
+            )
             new_key = key.replace("beta", "bias")
         if new_key:
             old_keys.append(key)
@@ -807,8 +815,16 @@ def _load_state_dict_into_meta_model(
     for key in state_dict.keys():
         new_key = None
         if "gamma" in key:
+            logger.warning(
+                "A parameter name that contains `gamma` will be renamed internally. Please use a different name "
+                "to suppress this warning."
+            )
             new_key = key.replace("gamma", "weight")
         if "beta" in key:
+            logger.warning(
+                "A parameter name that contains `beta` will be renamed internally. Please use a different name "
+                "to suppress this warning."
+            )
             new_key = key.replace("beta", "bias")
         if new_key:
             old_keys.append(key)
@@ -3980,8 +3996,16 @@ def _load_pretrained_model(
 
         def _fix_key(key):
             if "beta" in key:
+                logger.warning(
+                    "A parameter name that contains `beta` will be renamed internally. Please use a different name "
+                    "to suppress this warning."
+                )
                 return key.replace("beta", "bias")
             if "gamma" in key:
+                logger.warning(
+                    "A parameter name that contains `gamma` will be renamed internally. Please use a different name "
+                    "to suppress this warning."
+                )
                 return key.replace("gamma", "weight")
             return key
 

tests/utils/test_modeling_utils.py

@@ -1499,6 +1499,49 @@ def test_model_from_pretrained_from_mlx(self):
             outputs_from_saved = new_model(input_ids)
             self.assertTrue(torch.allclose(outputs_from_saved["logits"], outputs["logits"]))
 
+    def test_warning_for_beta_gamma_parameters(self):
+        class TestModelGamma(PreTrainedModel):
+            def __init__(self, config):
+                super().__init__(config)
+                self.gamma_param = nn.Parameter(torch.ones(10))
+                self.post_init()
+
+            def forward(self):
+                return self.gamma_param.sum()
+
+        logger = logging.get_logger("transformers.modeling_utils")
+        config = PretrainedConfig()
+        warning_msg_gamma = "A parameter name that contains `gamma` will be renamed internally"
+        model = TestModelGamma(config)
+
+        with tempfile.TemporaryDirectory() as tmp_dir:
+            model.save_pretrained(tmp_dir)
+            with LoggingLevel(logging.WARNING):
+                with CaptureLogger(logger) as cl1:
+                    TestModelGamma.from_pretrained(tmp_dir, config=config)
+
+        self.assertIn(warning_msg_gamma, cl1.out)
+
+        class TestModelBeta(PreTrainedModel):
+            def __init__(self, config):
+                super().__init__(config)
+                self.beta_param = nn.Parameter(torch.ones(10))
+                self.post_init()
+
+            def forward(self):
+                return self.beta_param.sum()
+
+        warning_msg_beta = "A parameter name that contains `beta` will be renamed internally"
+        model = TestModelBeta(config)
+
+        with tempfile.TemporaryDirectory() as tmp_dir:
+            model.save_pretrained(tmp_dir)
+            with LoggingLevel(logging.WARNING):
+                with CaptureLogger(logger) as cl2:
+                    TestModelBeta.from_pretrained(tmp_dir, config=config)
+
+        self.assertIn(warning_msg_beta, cl2.out)
+
 
 @slow
 @require_torch