Skip to content

Commit b5375c9

Browse files
committed
Add presets for gemma 2 2b (#1721)
1 parent 39802c7 commit b5375c9

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

keras_nlp/src/models/gemma/gemma_presets.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -155,6 +155,26 @@
155155
},
156156
"kaggle_handle": "kaggle://keras/codegemma/keras/code_gemma_1.1_instruct_7b_en/1",
157157
},
158+
"gemma2_2b_en": {
159+
"metadata": {
160+
"description": "2 billion parameter, 26-layer, base Gemma model.",
161+
"params": 2614341888,
162+
"official_name": "Gemma",
163+
"path": "gemma",
164+
"model_card": "https://www.kaggle.com/models/google/gemma",
165+
},
166+
"kaggle_handle": "kaggle://keras/gemma2/keras/gemma2_2b_en/1",
167+
},
168+
"gemma2_instruct_2b_en": {
169+
"metadata": {
170+
"description": "2 billion parameter, 26-layer, instruction tuned Gemma model.",
171+
"params": 2614341888,
172+
"official_name": "Gemma",
173+
"path": "gemma",
174+
"model_card": "https://www.kaggle.com/models/google/gemma",
175+
},
176+
"kaggle_handle": "kaggle://keras/gemma2/keras/gemma2_instruct_2b_en/1",
177+
},
158178
"gemma2_9b_en": {
159179
"metadata": {
160180
"description": "9 billion parameter, 42-layer, base Gemma model.",

0 commit comments

Comments
 (0)