Skip to content

Commit baafff0

Browse files
committed
remove accelerator count default
1 parent 467fffb commit baafff0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

eval_protocol/cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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")

0 commit comments

Comments
 (0)