Change name of the flag

This commit is contained in:
CrossProduct
2025-01-07 20:36:03 +00:00
parent eeea82ad04
commit f0db0ccb00
+1 -1
View File
@@ -455,7 +455,7 @@ def _add_training_arguments(parser: argparse.ArgumentParser) -> None:
"--training_type", "--training_type",
type=str, type=str,
default=None, 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("--seed", type=int, default=None, help="A seed for reproducible training.")
parser.add_argument( parser.add_argument(