Bump version to fix query norm in Gemma 2 9b (#1709)

keras-team · Jul 26, 2024 · 455c706 · 455c706
1 parent 6c66911
commit 455c706
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keras_nlp/src/models/gemma/gemma_presets.py b/keras_nlp/src/models/gemma/gemma_presets.py
@@ -163,7 +163,7 @@
             "path": "gemma",
             "model_card": "https://www.kaggle.com/models/google/gemma",
         },
-        "kaggle_handle": "kaggle://keras/gemma2/keras/gemma2_9b_en/1",
+        "kaggle_handle": "kaggle://keras/gemma2/keras/gemma2_9b_en/2",
     },
     "gemma2_instruct_9b_en": {
         "metadata": {
@@ -173,7 +173,7 @@
             "path": "gemma",
             "model_card": "https://www.kaggle.com/models/google/gemma",
         },
-        "kaggle_handle": "kaggle://keras/gemma2/keras/gemma2_instruct_9b_en/1",
+        "kaggle_handle": "kaggle://keras/gemma2/keras/gemma2_instruct_9b_en/2",
     },
     "gemma2_27b_en": {
         "metadata": {