Skip to content

Commit c73df25

Browse files
committed
Add Magistral Presets to Hub
1 parent 4136e38 commit c73df25

1 file changed

Lines changed: 18 additions & 0 deletions

File tree

keras_hub/src/models/mistral/mistral_presets.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
# Metadata for loading pretrained model weights.
44
backbone_presets = {
5+
# Mistral models
56
"mistral_7b_en": {
67
"metadata": {
78
"description": "Mistral 7B base model",
@@ -42,4 +43,21 @@
4243
},
4344
"kaggle_handle": "kaggle://keras/mistral/keras/mistral_0.3_instruct_7b_en/1",
4445
},
46+
# Magistral models
47+
"magistral_small_2506_en": {
48+
"metadata": {
49+
"description": "Magistral Small 2506 model",
50+
"params": 23572403200,
51+
"path": "mistral",
52+
},
53+
"kaggle_handle": "kaggle://keras/magistral/keras/magistral_small_2506_en/1",
54+
},
55+
"magistral_small_2507_en": {
56+
"metadata": {
57+
"description": "Magistral Small 2507 model",
58+
"params": 23572403200,
59+
"path": "mistral",
60+
},
61+
"kaggle_handle": "kaggle://keras/magistral/keras/magistral_small_2507_en/1",
62+
},
4563
}

0 commit comments

Comments
 (0)