From 5ab7cd4534958f82daf7e2b2aff98f7d50ceea5e Mon Sep 17 00:00:00 2001 From: Ryan Mullins Date: Tue, 5 Nov 2024 16:21:30 +0000 Subject: [PATCH] Correct model card links for Gemma variants --- keras_hub/src/models/gemma/gemma_presets.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/keras_hub/src/models/gemma/gemma_presets.py b/keras_hub/src/models/gemma/gemma_presets.py index 907dc4b13a..ecd22db5b0 100644 --- a/keras_hub/src/models/gemma/gemma_presets.py +++ b/keras_hub/src/models/gemma/gemma_presets.py @@ -47,7 +47,7 @@ "params": 2506172416, "official_name": "Gemma", "path": "gemma", - "model_card": "https://www.kaggle.com/models/google/gemma", + "model_card": "https://www.kaggle.com/models/google/codegemma", }, "kaggle_handle": "kaggle://keras/codegemma/keras/code_gemma_1.1_2b_en/1", }, @@ -61,7 +61,7 @@ "params": 2506172416, "official_name": "Gemma", "path": "gemma", - "model_card": "https://www.kaggle.com/models/google/gemma", + "model_card": "https://www.kaggle.com/models/google/codegemma", }, "kaggle_handle": "kaggle://keras/codegemma/keras/code_gemma_2b_en/1", }, @@ -110,7 +110,7 @@ "params": 8537680896, "official_name": "Gemma", "path": "gemma", - "model_card": "https://www.kaggle.com/models/google/gemma", + "model_card": "https://www.kaggle.com/models/google/codegemma", }, "kaggle_handle": "kaggle://keras/codegemma/keras/code_gemma_7b_en/1", }, @@ -124,7 +124,7 @@ "params": 8537680896, "official_name": "Gemma", "path": "gemma", - "model_card": "https://www.kaggle.com/models/google/gemma", + "model_card": "https://www.kaggle.com/models/google/codegemma", }, "kaggle_handle": "kaggle://keras/codegemma/keras/code_gemma_instruct_7b_en/1", }, @@ -138,7 +138,7 @@ "params": 8537680896, "official_name": "Gemma", "path": "gemma", - "model_card": "https://www.kaggle.com/models/google/gemma", + "model_card": "https://www.kaggle.com/models/google/codegemma", }, "kaggle_handle": "kaggle://keras/codegemma/keras/code_gemma_1.1_instruct_7b_en/1", }, @@ -208,7 +208,7 @@ "params": 2614341888, "official_name": "Gemma", "path": "gemma", - "model_card": "https://www.kaggle.com/models/google/gemma", + "model_card": "https://www.kaggle.com/models/google/shieldgemma", }, "kaggle_handle": "kaggle://google/shieldgemma/keras/shieldgemma_2b_en/1", }, @@ -218,7 +218,7 @@ "params": 9241705984, "official_name": "Gemma", "path": "gemma", - "model_card": "https://www.kaggle.com/models/google/gemma", + "model_card": "https://www.kaggle.com/models/google/shieldgemma", }, "kaggle_handle": "kaggle://google/shieldgemma/keras/shieldgemma_9b_en/1", }, @@ -228,7 +228,7 @@ "params": 27227128320, "official_name": "Gemma", "path": "gemma", - "model_card": "https://www.kaggle.com/models/google/gemma", + "model_card": "https://www.kaggle.com/models/google/shieldgemma", }, "kaggle_handle": "kaggle://google/shieldgemma/keras/shieldgemma_27b_en/1", },