diff --git a/ch06/01_main-chapter-code/load-finetuned-model.ipynb b/ch06/01_main-chapter-code/load-finetuned-model.ipynb
index fd7e180..f6d210d 100644
--- a/ch06/01_main-chapter-code/load-finetuned-model.ipynb
+++ b/ch06/01_main-chapter-code/load-finetuned-model.ipynb
@@ -124,7 +124,6 @@
     "BASE_CONFIG.update(model_configs[CHOOSE_MODEL])\n",
     "\n",
     "# Initialize base model\n",
-    "model_size = CHOOSE_MODEL.split(\" \")[-1].lstrip(\"(\").rstrip(\")\")\n",
     "model = GPTModel(BASE_CONFIG)"
    ]
   },