From 1544c4b340617147d81c3eea5211896c081c7934 Mon Sep 17 00:00:00 2001 From: RenuPatelGoogle <89264621+RenuPatelGoogle@users.noreply.github.com> Date: Wed, 25 Sep 2024 23:20:54 +0530 Subject: [PATCH] Corrected URL path to access the Kaggle Gemma model weights dir I have corrected incorrect URL for accessing the Kaggle Gemma-2 model checkpoint and weight directory from "google/gemma-2-2b/pyTorch/gemma-2-{VARIANT}" to "google/gemma-2/pyTorch/gemma-2-{VARIANT}". I have updated the URL after verifying on the Kaggle platform and also have successfully implemented and tested in a Colab environment. Please find the replicated [gist](https://colab.sandbox.google.com/gist/RenuPatelGoogle/711a744401b32b30048c2f100d0ab7d2/kaggle_gemma2.ipynb) here for the same. --- site/en/gemma/docs/pytorch_gemma.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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}')" ] }, {