Skip to content

Commit 7cc43fd

Browse files
authored
Add shortcuts for arguments specified in the FT docs (#279)
* Add shortcuts * Add shortcuts for download
1 parent ea39fb5 commit 7cc43fd

File tree

1 file changed

+21
-7
lines changed

1 file changed

+21
-7
lines changed

src/together/cli/api/finetune.py

Lines changed: 21 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -58,19 +58,27 @@ def fine_tuning(ctx: click.Context) -> None:
5858
@fine_tuning.command()
5959
@click.pass_context
6060
@click.option(
61-
"--training-file", type=str, required=True, help="Training file ID from Files API"
61+
"--training-file",
62+
"-t",
63+
type=str,
64+
required=True,
65+
help="Training file ID from Files API",
66+
)
67+
@click.option("--model", "-m", type=str, help="Base model name")
68+
@click.option(
69+
"--n-epochs", "-ne", type=int, default=1, help="Number of epochs to train for"
6270
)
63-
@click.option("--model", type=str, help="Base model name")
64-
@click.option("--n-epochs", type=int, default=1, help="Number of epochs to train for")
6571
@click.option(
6672
"--validation-file", type=str, default="", help="Validation file ID from Files API"
6773
)
6874
@click.option("--n-evals", type=int, default=0, help="Number of evaluation loops")
6975
@click.option(
70-
"--n-checkpoints", type=int, default=1, help="Number of checkpoints to save"
76+
"--n-checkpoints", "-c", type=int, default=1, help="Number of checkpoints to save"
7177
)
72-
@click.option("--batch-size", type=INT_WITH_MAX, default="max", help="Train batch size")
73-
@click.option("--learning-rate", type=float, default=1e-5, help="Learning rate")
78+
@click.option(
79+
"--batch-size", "-b", type=INT_WITH_MAX, default="max", help="Train batch size"
80+
)
81+
@click.option("--learning-rate", "-lr", type=float, default=1e-5, help="Learning rate")
7482
@click.option(
7583
"--min-lr-ratio",
7684
type=float,
@@ -123,7 +131,11 @@ def fine_tuning(ctx: click.Context) -> None:
123131
help="Beta parameter for DPO training (only used when '--training-method' is 'dpo')",
124132
)
125133
@click.option(
126-
"--suffix", type=str, default=None, help="Suffix for the fine-tuned model name"
134+
"--suffix",
135+
"-s",
136+
type=str,
137+
default=None,
138+
help="Suffix for the fine-tuned model name",
127139
)
128140
@click.option("--wandb-api-key", type=str, default=None, help="Wandb API key")
129141
@click.option("--wandb-base-url", type=str, default=None, help="Wandb base URL")
@@ -418,13 +430,15 @@ def list_checkpoints(ctx: click.Context, fine_tune_id: str) -> None:
418430
@click.argument("fine_tune_id", type=str, required=True)
419431
@click.option(
420432
"--output_dir",
433+
"-o",
421434
type=click.Path(exists=True, file_okay=False, resolve_path=True),
422435
required=False,
423436
default=None,
424437
help="Output directory",
425438
)
426439
@click.option(
427440
"--checkpoint-step",
441+
"-s",
428442
type=int,
429443
required=False,
430444
default=None,

0 commit comments

Comments
 (0)