File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed
Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -402,7 +402,7 @@ def parse_args(args=None):
402402 rft_parser .add_argument ("--lora-rank" , type = int , default = 16 )
403403 rft_parser .add_argument ("--gradient-accumulation-steps" , type = int , help = "Number of gradient accumulation steps" )
404404 rft_parser .add_argument ("--learning-rate-warmup-steps" , type = int , help = "Number of LR warmup steps" )
405- rft_parser .add_argument ("--accelerator-count" , type = int , default = 1 )
405+ rft_parser .add_argument ("--accelerator-count" , type = int )
406406 rft_parser .add_argument ("--region" , help = "Fireworks region enum value" )
407407 rft_parser .add_argument ("--display-name" , help = "RFT job display name" )
408408 rft_parser .add_argument ("--evaluation-dataset" , help = "Optional separate eval dataset id" )
You can’t perform that action at this time.
0 commit comments