@@ -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