diff --git a/src/axolotl/core/trainer_builder/base.py b/src/axolotl/core/trainer_builder/base.py index 6c3735256..90d6a2d9d 100644 --- a/src/axolotl/core/trainer_builder/base.py +++ b/src/axolotl/core/trainer_builder/base.py @@ -421,6 +421,10 @@ class TrainerBuilderBase(abc.ABC): adam_kwargs["eps"] = (eps1, eps2) optimizer_kwargs.update(adam_kwargs) + else: + raise ValueError( + f"Unhandled optimizer: {self.cfg.optimizer}. Please raise an Issue." + ) # Parse any additional optimizer args from config if self.cfg.optim_args: