add general arguments to train as well

2025-03-03 19:21:01 -08:00 · 2025-03-03 19:21:01 -08:00 · 50734795b9
commit 50734795b9
parent bf1e366e43
1 changed files with 1 additions and 0 deletions
--- 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()