diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py index 2d7020ffafbd..1002ceb0cfe2 100755 --- a/src/transformers/trainer.py +++ b/src/transformers/trainer.py @@ -1274,7 +1274,7 @@ def get_optimizer_cls_and_kwargs( optimizer_kwargs.update(additional_optim_kwargs) optimizer_kwargs.update(bnb_kwargs) except ImportError: - raise ValueError("Trainer tried to instantiate bnb optimizer but bnb is not installed!") + raise ValueError("Trainer tried to instantiate bnb optimizer but `bitsandbytes` is not installed!") if is_bitsandbytes_available() and version.parse( importlib.metadata.version("bitsandbytes") ) < version.parse("0.41.1"): diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py index 00d15acf18be..a557e0f80b7b 100644 --- a/tests/trainer/test_trainer.py +++ b/tests/trainer/test_trainer.py @@ -4427,7 +4427,7 @@ def test_bnb_adam8bit_no_bnb(self): args = TrainingArguments(optim=OptimizerNames.ADAMW_BNB, output_dir="None") # Pretend that bnb does not exist, even if installed. By setting bnb to None, importing - # bnb will fail even if bnb is installed. + # bnb will fail even if `bitsandbytes` is installed. with patch.dict("sys.modules", {"bitsandbytes.optim": None}): with self.assertRaises(ValueError): Trainer.get_optimizer_cls_and_kwargs(args) @@ -4436,7 +4436,7 @@ def test_bnb_paged_adam_no_bnb(self): args = TrainingArguments(optim=OptimizerNames.PAGED_ADAMW, output_dir="None") # Pretend that bnb does not exist, even if installed. By setting bnb to None, importing - # bnb will fail even if bnb is installed. + # bnb will fail even if `bitsandbytes` is installed. with patch.dict("sys.modules", {"bitsandbytes.optim": None}): with self.assertRaises(ValueError): Trainer.get_optimizer_cls_and_kwargs(args) @@ -4445,7 +4445,7 @@ def test_bnb_paged_adam8bit_no_bnb(self): args = TrainingArguments(optim=OptimizerNames.PAGED_ADAMW_8BIT, output_dir="None") # Pretend that bnb does not exist, even if installed. By setting bnb to None, importing - # bnb will fail even if bnb is installed. + # bnb will fail even if `bitsandbytes` is installed. with patch.dict("sys.modules", {"bitsandbytes.optim": None}): with self.assertRaises(ValueError): Trainer.get_optimizer_cls_and_kwargs(args) @@ -4454,7 +4454,7 @@ def test_bnb_paged_lion_no_bnb(self): args = TrainingArguments(optim=OptimizerNames.PAGED_LION, output_dir="None") # Pretend that bnb does not exist, even if installed. By setting bnb to None, importing - # bnb will fail even if bnb is installed. + # bnb will fail even if `bitsandbytes` is installed. with patch.dict("sys.modules", {"bitsandbytes.optim": None}): with self.assertRaises(ValueError): Trainer.get_optimizer_cls_and_kwargs(args) @@ -4463,7 +4463,7 @@ def test_bnb_paged_lion8bit_no_bnb(self): args = TrainingArguments(optim=OptimizerNames.PAGED_LION_8BIT, output_dir="None") # Pretend that bnb does not exist, even if installed. By setting bnb to None, importing - # bnb will fail even if bnb is installed. + # bnb will fail even if `bitsandbytes` is installed. with patch.dict("sys.modules", {"bitsandbytes.optim": None}): with self.assertRaises(ValueError): Trainer.get_optimizer_cls_and_kwargs(args)