From 50734795b93919d25cfe7159836c6309da8b25af Mon Sep 17 00:00:00 2001 From: Dana Case Date: Mon, 3 Mar 2025 19:21:01 -0800 Subject: [PATCH] add general arguments to train as well --- src/mflux/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/mflux/train.py b/src/mflux/train.py index 1d035c1..a8f07ae 100644 --- a/src/mflux/train.py +++ b/src/mflux/train.py @@ -6,6 +6,7 @@ from mflux.ui.cli.parsers import CommandLineParser def main(): parser = CommandLineParser(description="Finetune a LoRA adapter") + parser.add_general_arguments() parser.add_model_arguments(require_model_arg=False) parser.add_training_arguments() args = parser.parse_args()