fix some errors due to inconsistency of model cards

Former-commit-id: dd83265b9b8768eb8732f59ace128dfe4aac1c47
2024-09-30 19:58:34 +08:00
parent 32b5364051
commit 8fd84c375e
5 changed files with 66 additions and 69 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -1164,12 +1164,12 @@ register_model_group(

 register_model_group(
    models={
-        "Pixtral-12B": {
+        "Pixtral-12B-Instruct": {
            DownloadSource.DEFAULT: "mistral-community/pixtral-12b",
            DownloadSource.MODELSCOPE: "AI-ModelScope/pixtral-12b",
        }
    },
-    template="mistral",
+    template="pixtral",
    vision=True
 )