Update to the newest version of Gemma on Kaggle (#1454)

Includes some small cleanups for the Kaggle assets.
keras-team · Feb 22, 2024 · e5a91f7 · e5a91f7
1 parent dd3ceb7
commit e5a91f7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/keras_nlp/models/gemma/gemma_presets.py b/keras_nlp/models/gemma/gemma_presets.py
@@ -25,7 +25,7 @@
             "path": "gemma",
             "model_card": "https://www.kaggle.com/models/google/gemma",
         },
-        "kaggle_handle": "kaggle://keras/gemma/keras/gemma_2b_en/1",
+        "kaggle_handle": "kaggle://keras/gemma/keras/gemma_2b_en/2",
     },
     "gemma_instruct_2b_en": {
         "metadata": {
@@ -37,7 +37,7 @@
             "path": "gemma",
             "model_card": "https://www.kaggle.com/models/google/gemma",
         },
-        "kaggle_handle": "kaggle://keras/gemma/keras/gemma_instruct_2b_en/1",
+        "kaggle_handle": "kaggle://keras/gemma/keras/gemma_instruct_2b_en/2",
     },
     "gemma_7b_en": {
         "metadata": {
@@ -49,7 +49,7 @@
             "path": "gemma",
             "model_card": "https://www.kaggle.com/models/google/gemma",
         },
-        "kaggle_handle": "kaggle://keras/gemma/keras/gemma_7b_en/1",
+        "kaggle_handle": "kaggle://keras/gemma/keras/gemma_7b_en/2",
     },
     "gemma_instruct_7b_en": {
         "metadata": {
@@ -61,6 +61,6 @@
             "path": "gemma",
             "model_card": "https://www.kaggle.com/models/google/gemma",
         },
-        "kaggle_handle": "kaggle://keras/gemma/keras/gemma_instruct_7b_en/1",
+        "kaggle_handle": "kaggle://keras/gemma/keras/gemma_instruct_7b_en/2",
     },
 }