Skip to content

Commit

Permalink
Corrected URL path to access the Kaggle Gemma model weights dir
Browse files Browse the repository at this point in the history
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.
  • Loading branch information
RenuPatelGoogle authored Sep 25, 2024
1 parent 205547a commit 1544c4b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion site/en/gemma/docs/pytorch_gemma.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -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}')"
]
},
{
Expand Down

0 comments on commit 1544c4b

Please sign in to comment.