diff --git a/README.md b/README.md index 8f2e302..8e601ba 100644 --- a/README.md +++ b/README.md @@ -111,7 +111,7 @@ ft_cfg = OpenAIFTJob( ```bash python scripts/run_finetuning_job.py \ --config_module=cfgs/preference_numbers/cfgs.py \ - --cfg_var_name=animal_evaluation \ + --cfg_var_name=ft_job_cfg \ --dataset_path=./data/preference_numbers/owl/filtered_dataset.jsonl \ --output_path=./data/preference_numbers/owl/model.json ``` diff --git a/sl/finetuning/services.py b/sl/finetuning/services.py index f230ede..3dc442d 100644 --- a/sl/finetuning/services.py +++ b/sl/finetuning/services.py @@ -126,7 +126,7 @@ async def _run_openai_finetuning_job( # Create fine-tuning job client = openai_driver.get_client() oai_job = await client.fine_tuning.jobs.create( - model=cfg.source_model_id, + model=cfg.source_model.id, training_file=file_obj.id, method=Method( type="supervised", @@ -159,8 +159,8 @@ async def _run_openai_finetuning_job( # Wait before polling again await asyncio.sleep(30) - assert oai_job.fine_tuned_model is not None - return Model(id=oai_job.fine_tuned_model, type="openai") + assert job_status.fine_tuned_model is not None + return Model(id=job_status.fine_tuned_model, type="openai") async def run_finetuning_job(job: FTJob, dataset: list[DatasetRow]) -> Model: