From f0db0ccb009741c82d24547ee4bcdc1391f93b3f Mon Sep 17 00:00:00 2001 From: CrossProduct Date: Tue, 7 Jan 2025 20:36:03 +0000 Subject: [PATCH] Change name of the flag --- finetrainers/args.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/finetrainers/args.py b/finetrainers/args.py index 5ca21b5..224b9f1 100644 --- a/finetrainers/args.py +++ b/finetrainers/args.py @@ -455,7 +455,7 @@ def _add_training_arguments(parser: argparse.ArgumentParser) -> None: "--training_type", type=str, default=None, - help="Type of training to perform. Choose between ['lora','finetune']", + help="Type of training to perform. Choose between ['lora','full_finetune']", ) parser.add_argument("--seed", type=int, default=None, help="A seed for reproducible training.") parser.add_argument(