fixed bug in test_finetuned notebook to read from proper directory

test_finetuned.ipynb

@@ -58,7 +58,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -78,8 +78,8 @@
     "print(device)\n",
     "\n",
     "# Load model and tokenizer\n",
-    "tokenizer = AutoTokenizer.from_pretrained(\"./fine-tuned-model\")\n",
-    "model = AutoModelForCausalLM.from_pretrained(\"./fine-tuned-model\", torch_dtype=torch.bfloat16, device_map=device) \n",
+    "tokenizer = AutoTokenizer.from_pretrained(\"./fine-tuned-model-16\")\n",
+    "model = AutoModelForCausalLM.from_pretrained(\"./fine-tuned-model-16\", torch_dtype=torch.bfloat16, device_map=device) \n",
     "model.generation_config.pad_token_id = tokenizer.pad_token_id"
    ]
   },