Skip to content

Commit 7e40a56

Browse files
fix undefined variables in fine tune example (#660)
`create_user_message` and `test_df` are not defined.
1 parent c6c92ac commit 7e40a56

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

examples/How_to_finetune_chat_models.ipynb

+5-1
Original file line numberDiff line numberDiff line change
@@ -253,11 +253,14 @@
253253
"\n",
254254
"system_message = \"You are a helpful recipe assistant. You are to extract the generic ingredients from each of the recipes provided.\"\n",
255255
"\n",
256+
"def create_user_message(row):\n",
257+
" return f\"\"\"Title: {row['title']}\\n\\nIngredients: {row['ingredients']}\\n\\nGeneric ingredients: \"\"\"\n",
258+
"\n",
256259
"def prepare_example_conversation(row):\n",
257260
" messages = []\n",
258261
" messages.append({\"role\": \"system\", \"content\": system_message})\n",
259262
"\n",
260-
" user_message = f\"\"\"Title: {row['title']}\\n\\nIngredients: {row['ingredients']}\\n\\nGeneric ingredients: \"\"\"\n",
263+
" user_message = create_user_message(row)\n",
261264
" messages.append({\"role\": \"user\", \"content\": user_message})\n",
262265
"\n",
263266
" messages.append({\"role\": \"assistant\", \"content\": row[\"NER\"]})\n",
@@ -642,6 +645,7 @@
642645
}
643646
],
644647
"source": [
648+
"test_df = recipe_df.loc[201:300]\n",
645649
"test_row = test_df.iloc[0]\n",
646650
"test_messages = []\n",
647651
"test_messages.append({\"role\": \"system\", \"content\": system_message})\n",

0 commit comments

Comments
 (0)