fix: force access to lr & weight decay in case not provided to early error

2025-05-23 12:31:02 +07:00
parent 255acd3da2
commit 7496db524b
1 changed files with 2 additions and 2 deletions
--- a/src/axolotl/core/trainer_builder/base.py
+++ b/src/axolotl/core/trainer_builder/base.py
@@ -237,8 +237,8 @@ class TrainerBuilderBase(abc.ABC):
        if self.cfg.optimizer in custom_supported_optimizers:
            # Common optimizer kwargs
            optimizer_kwargs = {
-                "lr": training_args_kwargs.get("learning_rate"),
-                "weight_decay": training_args_kwargs.get("weight_decay"),
+                "lr": training_args_kwargs["learning_rate"],
+                "weight_decay": training_args_kwargs["weight_decay"],
            }

            # Adam-specific kwargs