diff --git a/fine_tuning_gemma2.ipynb b/fine_tuning_gemma2.ipynb
index 29348a2..d3341be 100644
--- a/fine_tuning_gemma2.ipynb
+++ b/fine_tuning_gemma2.ipynb
@@ -2968,14 +2968,14 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 15,
    "metadata": {},
    "outputs": [
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Loading checkpoint shards: 100%|██████████| 2/2 [00:01<00:00,  1.26it/s]\n"
+      "Loading checkpoint shards: 100%|██████████| 2/2 [00:05<00:00,  2.64s/it]\n"
      ]
     }
    ],
@@ -2983,7 +2983,7 @@
     "from transformers import AutoTokenizer, AutoConfig, AutoModelForCausalLM\n",
     "from latent_gemma import LatentReasoningGemmaForCausalLM\n",
     "\n",
-    "model_path = \"/home/featurize/.cache/kagglehub/models/victorumesiobi/gemma-2-japanese-english-reasoning/transformers/1/1/transformers\" # Replace with the path to which your model was downloaded too\n",
+    "model_path = \"/home/featurize/.cache/kagglehub/models/victorumesiobi/gemma-2-japanese-english-reasoning/transformers/1/2\" # Replace with the path to which your model was downloaded too\n",
     "\n",
     "# Load model and tokenizer\n",
     "tokenizer = AutoTokenizer.from_pretrained(model_path)\n",
@@ -3044,7 +3044,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": 14,
    "metadata": {},
    "outputs": [
     {