diff --git a/keras_nlp/models/gemma/gemma_presets.py b/keras_nlp/models/gemma/gemma_presets.py index 62d57344c2..a65a6da43a 100644 --- a/keras_nlp/models/gemma/gemma_presets.py +++ b/keras_nlp/models/gemma/gemma_presets.py @@ -48,7 +48,21 @@ "path": "gemma", "model_card": "https://www.kaggle.com/models/google/gemma", }, - "kaggle_handle": "kaggle://keras/gemma/keras/gemma_1.1_instruct_2b_en/1", + "kaggle_handle": "kaggle://keras/gemma/keras/gemma_1.1_instruct_2b_en/3", + }, + "code_gemma_2b_en": { + "metadata": { + "description": ( + "2 billion parameter, 18-layer, CodeGemma model. This model " + "has been trained on a fill-in-the-middle (FIM) task for code " + "completion." + ), + "params": 2506172416, + "official_name": "Gemma", + "path": "gemma", + "model_card": "https://www.kaggle.com/models/google/gemma", + }, + "kaggle_handle": "kaggle://keras/codegemma/keras/code_gemma_2b_en/1", }, "gemma_7b_en": { "metadata": { @@ -83,6 +97,34 @@ "path": "gemma", "model_card": "https://www.kaggle.com/models/google/gemma", }, - "kaggle_handle": "kaggle://keras/gemma/keras/gemma_1.1_instruct_7b_en/1", + "kaggle_handle": "kaggle://keras/gemma/keras/gemma_1.1_instruct_7b_en/3", + }, + "code_gemma_7b_en": { + "metadata": { + "description": ( + "7 billion parameter, 28-layer, CodeGemma model. This model " + "has been trained on a fill-in-the-middle (FIM) task for code " + "completion." + ), + "params": 8537680896, + "official_name": "Gemma", + "path": "gemma", + "model_card": "https://www.kaggle.com/models/google/gemma", + }, + "kaggle_handle": "kaggle://keras/codegemma/keras/code_gemma_7b_en/1", + }, + "code_gemma_instruct_7b_en": { + "metadata": { + "description": ( + "7 billion parameter, 28-layer, instruction tuned CodeGemma " + "model. This model has been trained for chat use cases related " + "to code." + ), + "params": 8537680896, + "official_name": "Gemma", + "path": "gemma", + "model_card": "https://www.kaggle.com/models/google/gemma", + }, + "kaggle_handle": "kaggle://keras/codegemma/keras/code_gemma_instruct_7b_en/1", }, }