From e5a91f7abc0ab9eed869f29451e74daaded48895 Mon Sep 17 00:00:00 2001
From: Matt Watson <1389937+mattdangerw@users.noreply.github.com>
Date: Wed, 21 Feb 2024 17:08:30 -0800
Subject: [PATCH] Update to the newest version of Gemma on Kaggle (#1454)

Includes some small cleanups for the Kaggle assets.
---
 keras_nlp/models/gemma/gemma_presets.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/keras_nlp/models/gemma/gemma_presets.py b/keras_nlp/models/gemma/gemma_presets.py
index f63fef17fa..72360f72dc 100644
--- 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",
     },
 }