Skip to content

Commit 4ef63e6

Browse files
authored
Enable newly released Med Gemma 1.5 4B variant version to hub (keras-team#2524)
* Enable newly released medgemma 1.5 version to hub * Fix Format issue * Update presets description * Fix Format
1 parent 2909d49 commit 4ef63e6

1 file changed

Lines changed: 31 additions & 7 deletions

File tree

keras_hub/src/models/gemma3/gemma3_presets.py

Lines changed: 31 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -181,12 +181,25 @@
181181
},
182182
"kaggle_handle": "kaggle://keras/gemma3/keras/gemma3_instruct_270m/4",
183183
},
184+
"medgemma_4b": {
185+
"metadata": {
186+
"description": (
187+
"A 4 billion parameter model based on Gemma 3. "
188+
"This model is pre-trained for performance on medical text "
189+
"and image comprehension and is optimized for medical "
190+
"applications that involve a text generation component."
191+
),
192+
"params": 4300079472,
193+
"path": "gemma3",
194+
},
195+
"kaggle_handle": "kaggle://keras/medgemma/keras/medgemma_4b/1",
196+
},
184197
"medgemma_instruct_4b": {
185198
"metadata": {
186199
"description": (
187200
"A 4 billion parameter model based on Gemma 3. "
188-
"This model is trained for performance on medical text"
189-
"and image comprehension and is optimized for medical"
201+
"This model is instruction-tuned for performance on medical "
202+
"text and image comprehension and is optimized for medical "
190203
"applications that involve a text generation component."
191204
),
192205
"params": 4300079472,
@@ -198,8 +211,8 @@
198211
"metadata": {
199212
"description": (
200213
"A 27 billion parameter model based on Gemma 3. "
201-
"This model trained for performance on medical text "
202-
"and image comprehension and is optimized for medical "
214+
"This model is instruction-tuned for performance on medical "
215+
" text and image comprehension and is optimized for medical "
203216
"applications that involve a text generation component."
204217
),
205218
"params": 27432406640,
@@ -211,15 +224,26 @@
211224
"metadata": {
212225
"description": (
213226
"A 27 billion parameter text-only model based on Gemma 3. "
214-
"This model is trained for performance on medical text "
215-
"comprehension and is optimized for medical applications "
216-
"that involve a text generation component."
227+
"This model is instruction-tuned (No images) for performance "
228+
"on medical text comprehension and is optimized for medical "
229+
"applications that involve a text generation component."
217230
),
218231
"params": 27009002240,
219232
"path": "gemma3",
220233
},
221234
"kaggle_handle": "kaggle://keras/medgemma/keras/medgemma_instruct_27b_text/1",
222235
},
236+
"medgemma_1.5_instruct_4b": {
237+
"metadata": {
238+
"description": (
239+
"A 4 billion parameter,Instruct-tuned MedGemma 1.5 4B is an "
240+
"updated version of the Instruction-tuned MedGemma 4B model."
241+
),
242+
"params": 4300079472,
243+
"path": "gemma3",
244+
},
245+
"kaggle_handle": "kaggle://keras/medgemma/keras/medgemma_1.5_instruct_4b/1",
246+
},
223247
"function_gemma_instruct_270m": {
224248
"metadata": {
225249
"description": (

0 commit comments

Comments
 (0)