add cookbook finetune_with_bos

baidubce · Feb 2, 2024 · 52e2e2f · 52e2e2f
1 parent ee4674b
commit 52e2e2f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cookbook/finetune/finetune_with_bos_and_evaluate.ipynb b/cookbook/finetune/finetune_with_bos_and_evaluate.ipynb
@@ -116,7 +116,7 @@
     "bos_ds = Dataset.load(bds,file_format=FormatType.Jsonl)\n",
     "\n",
     "trainer = LLMFinetune(\n",
-    "    train_type=\"ERNIE-Bot-turbo-0725\",\n",
+    "    train_type=\"ERNIE-Speed\",\n",
     "    # dataset_bos_path=f\"bos:/{bos_bucket_name}/{bos_bucket_file_path}\",\n",
     "    dataset=bos_ds\n",
     ")"
@@ -199,7 +199,8 @@
    "source": [
     "from qianfan.model import Model\n",
     "\n",
-    "m: Model = Model(version_id=\"amv-vge90genq6ed\")\n",
+    "m = trainer.result[0]['model']\n",
+    "# m = Model(version_id=\"amv-vge90genq6ed\")\n",
     "m.auto_complete_info()"
    ]
   },