fix

Author: rasbt

ch03/01_main-chapter-code/ch03_main.ipynb

@@ -220,7 +220,7 @@
     "# device = torch.device(\"cpu\")\n",
     "\n",
     "model, tokenizer = load_model_and_tokenizer(\n",
-    "    which_model=which_model,\n",
+    "    which_model=WHICH_MODEL,\n",
     "    device=device,\n",
     "    use_compile=False\n",
     ")"