From 981a1506613bbf2b007197149c0e78f1572a20c6 Mon Sep 17 00:00:00 2001 From: Samaneh Saadat Date: Wed, 10 Apr 2024 17:05:57 +0000 Subject: [PATCH] Revert "Update presets for code gemma (#1564)" This reverts commit 751fbfa722e3e364d80d4c62141fe4e6887d8b58. --- keras_nlp/models/gemma/gemma_presets.py | 46 ++----------------------- 1 file changed, 2 insertions(+), 44 deletions(-) diff --git a/keras_nlp/models/gemma/gemma_presets.py b/keras_nlp/models/gemma/gemma_presets.py index a65a6da43a..62d57344c2 100644 --- a/keras_nlp/models/gemma/gemma_presets.py +++ b/keras_nlp/models/gemma/gemma_presets.py @@ -48,21 +48,7 @@ "path": "gemma", "model_card": "https://www.kaggle.com/models/google/gemma", }, - "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", + "kaggle_handle": "kaggle://keras/gemma/keras/gemma_1.1_instruct_2b_en/1", }, "gemma_7b_en": { "metadata": { @@ -97,34 +83,6 @@ "path": "gemma", "model_card": "https://www.kaggle.com/models/google/gemma", }, - "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", + "kaggle_handle": "kaggle://keras/gemma/keras/gemma_1.1_instruct_7b_en/1", }, }