Skip to content

Commit 3f904c3

Browse files
Arsh ZahedArsh Zahed
authored andcommitted
Add shortcuts
1 parent b24a5c3 commit 3f904c3

1 file changed

Lines changed: 19 additions & 7 deletions

File tree

src/together/cli/api/finetune.py

Lines changed: 19 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, required=True, 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, required=True, 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")

0 commit comments

Comments
 (0)