[model] support Mistral3.1 small 2503 (#8335)

2025-06-09 10:37:42 +08:00
parent fa4360dca7
commit 31bca4d172
5 changed files with 30 additions and 4 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -1592,6 +1592,22 @@ register_model_group(
 )


+register_model_group(
+    models={
+        "Mistral-Small-24B-Base-2503": {
+            DownloadSource.DEFAULT: "mistralai/Mistral-Small-24B-Base-2503",
+            DownloadSource.MODELSCOPE: "mistralai/Mistral-Small-24B-Base-2503",
+        },
+        "Mistral-Small-24B-Instruct-2503": {
+            DownloadSource.DEFAULT: "mistralai/Mistral-Small-24B-Instruct-2503",
+            DownloadSource.MODELSCOPE: "mistralai/Mistral-Small-24B-Instruct-2503",
+        },
+    },
+    template="mistral_small",
+    multimodal=True,
+)
+
+
 register_model_group(
    models={
        "Mixtral-8x7B-v0.1": {