use default value for temperature

2026-04-10 12:33:42 +00:00 · 2024-05-19 08:48:10 -05:00
parent faffebae4b
commit 59f5ed8d68
3 changed files with 3 additions and 3 deletions
--- a/ch05/01_main-chapter-code/ch05.ipynb
+++ b/ch05/01_main-chapter-code/ch05.ipynb
@@ -1852,7 +1852,7 @@
   "metadata": {},
   "outputs": [],
   "source": [
-    "def generate(model, idx, max_new_tokens, context_size, temperature, top_k=None, eos_id=None):\n",
+    "def generate(model, idx, max_new_tokens, context_size, temperature=1.0, top_k=None, eos_id=None):\n",
    "\n",
    "    # For-loop is the same as before: Get logits, and only focus on last time step\n",
    "    for _ in range(max_new_tokens):\n",