diff --git a/site/en/gemma/docs/pytorch_gemma.ipynb b/site/en/gemma/docs/pytorch_gemma.ipynb
index 6f1cbdfe2..418c5ca3c 100644
--- a/site/en/gemma/docs/pytorch_gemma.ipynb
+++ b/site/en/gemma/docs/pytorch_gemma.ipynb
@@ -186,7 +186,7 @@
         "import kagglehub\n",
         "\n",
         "# Load model weights\n",
-        "weights_dir = kagglehub.model_download(f'google/gemma-2-2b/pyTorch/gemma-2-{VARIANT}')"
+        "weights_dir = kagglehub.model_download(f'google/gemma-2/pyTorch/gemma-2-{VARIANT}')"
       ]
     },
     {