lengths cleanup

lengths.ipynb

@@ -53,14 +53,16 @@
     "### Input:\n",
     "{data_point[\"input\"]}\n",
     "\n",
-    "### Response:\"\"\"\n",
+    "### Response:\n",
+    "{data_point[\"output\"]}\"\"\"\n",
     "    else:\n",
     "        return f\"\"\"Below is an instruction that describes a task. Write a response that appropriately completes the request.\n",
     "\n",
     "### Instruction:\n",
     "{data_point[\"instruction\"]}\n",
     "\n",
-    "### Response:\"\"\"\n",
+    "### Response:\n",
+    "{data_point[\"output\"]}\"\"\"\n",
     "\n",
     "\n",
     "data = data.map(lambda data_point: {\"prompt\": tokenizer(generate_prompt(data_point))})"