Skip to content

Commit e15d855

Browse files
committed
auto generate output model id
1 parent 1e63fcd commit e15d855

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

eval_protocol/cli_commands/create_rft.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
)
1414
from ..fireworks_rft import (
1515
_map_api_host_to_app_host,
16+
build_default_output_model,
1617
create_dataset_from_jsonl,
1718
create_reinforcement_fine_tuning_job,
1819
)
@@ -467,6 +468,10 @@ def create_rft_command(args) -> int:
467468
body["evaluationDataset"] = args.evaluation_dataset
468469
if getattr(args, "output_model", None):
469470
body.setdefault("trainingConfig", {})["outputModel"] = f"accounts/{account_id}/models/{args.output_model}"
471+
else:
472+
# Auto-generate output model name if not provided
473+
auto_output_model = build_default_output_model(evaluator_id)
474+
body.setdefault("trainingConfig", {})["outputModel"] = f"accounts/{account_id}/models/{auto_output_model}"
470475

471476
# Clean None fields to avoid noisy payloads
472477
body = {k: v for k, v in body.items() if v is not None}

0 commit comments

Comments
 (0)