diff --git a/llms/mlx_lm/lora.py b/llms/mlx_lm/lora.py index 2c869467..042b40e2 100644 --- a/llms/mlx_lm/lora.py +++ b/llms/mlx_lm/lora.py @@ -104,7 +104,7 @@ def build_parser(): "--optimizer", type=str, choices=["adam", "adamw"], - default="adam", + default=None, help="Optimizer to use for training: adam or adamw", ) parser.add_argument(