Skip to content

Commit b701bc0

Browse files
committed
Small fixes for reading the input files correctly and to the OpenAIFineTuner
1 parent 5d64d5d commit b701bc0

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

examples/fine-tuned_qa/ft_retrieval_augmented_generation_qdrant.ipynb

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -559,7 +559,7 @@
559559
"\n",
560560
" def create_openai_file(self):\n",
561561
" self.file_object = client.files.create(\n",
562-
" file=open(self.training_file_path, \"r\"),\n",
562+
" file=open(self.training_file_path, \"rb\"),\n",
563563
" purpose=\"fine-tune\",\n",
564564
" )\n",
565565
"\n",
@@ -571,19 +571,22 @@
571571
"\n",
572572
" def create_fine_tuning_job(self):\n",
573573
" self.fine_tuning_job = client.fine_tuning.jobs.create(\n",
574-
" training_file=self.file_object[\"id\"],\n",
574+
" training_file=self.file_object.id,\n",
575575
" model=self.model_name,\n",
576576
" suffix=self.suffix,\n",
577577
" )\n",
578578
"\n",
579579
" def wait_for_fine_tuning(self, sleep_time=45):\n",
580-
" while self.fine_tuning_job.status != 'succeeded':\n",
580+
" while True:\n",
581+
" # Retrieve the latest fine-tuning job status\n",
582+
" self.fine_tuning_job = client.fine_tuning.jobs.retrieve(self.fine_tuning_job.id)\n",
583+
" print(\"Job Status:\", self.fine_tuning_job.status)\n",
584+
" if self.fine_tuning_job.status in {'succeeded', 'failed', 'cancelled'}:\n",
585+
" break\n",
581586
" time.sleep(sleep_time)\n",
582-
" self.fine_tuning_job.refresh()\n",
583-
" print(\"Job Status: \", self.fine_tuning_job.status)\n",
584587
"\n",
585588
" def retrieve_fine_tuned_model(self):\n",
586-
" self.model_id = client.fine_tuning.jobs.retrieve(self.fine_tuning_job[\"id\"]).fine_tuned_model\n",
589+
" self.model_id = client.fine_tuning.jobs.retrieve(self.fine_tuning_job.id).fine_tuned_model\n",
587590
" return self.model_id\n",
588591
"\n",
589592
" def fine_tune_model(self):\n",

0 commit comments

Comments
 (0)