Skip to content

Commit e74791e

Browse files
authored
Register Cell2Sentence Presets (#2442)
* Register Cell2Sentence presets * Address gemini comments
1 parent 165c023 commit e74791e

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

keras_hub/src/models/gemma/gemma_presets.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -206,4 +206,26 @@
206206
},
207207
"kaggle_handle": "kaggle://keras/vaultgemma/keras/vault_gemma_1b_en/2",
208208
},
209+
"c2s_scale_gemma_2_2b_en": {
210+
"metadata": {
211+
"description": (
212+
"A 2 billion parameter, single-cell biology-aware model "
213+
"built on the Gemma-2 architecture."
214+
),
215+
"params": 2614341888,
216+
"path": "gemma",
217+
},
218+
"kaggle_handle": "kaggle://keras/cell2sentence/keras/c2s_scale_gemma_2_2b_en/1",
219+
},
220+
"c2s_scale_gemma_2_27b_en": {
221+
"metadata": {
222+
"description": (
223+
"A 27 billion parameter, single-cell biology-aware model "
224+
"built on the Gemma-2 architecture."
225+
),
226+
"params": 27227128320,
227+
"path": "gemma",
228+
},
229+
"kaggle_handle": "kaggle://keras/cell2sentence/keras/c2s_scale_gemma_2_27b_en/1",
230+
},
209231
}

0 commit comments

Comments
 (0)