From 455c70649d2ac4cf2ae54c1cd6ce28abe517928b Mon Sep 17 00:00:00 2001
From: Matt Watson <1389937+mattdangerw@users.noreply.github.com>
Date: Fri, 26 Jul 2024 10:18:02 -0700
Subject: [PATCH] Bump version to fix query norm in Gemma 2 9b (#1709)

---
 keras_nlp/src/models/gemma/gemma_presets.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/keras_nlp/src/models/gemma/gemma_presets.py b/keras_nlp/src/models/gemma/gemma_presets.py
index 266619673..0ef91a346 100644
--- 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": {