remove some unnecessary if conditions

Former-commit-id: 482d3e5ff3338385da664475fee88c7dc623c993
2024-09-28 02:14:06 +08:00
parent 9390927875
commit e4c57f54f8
3 changed files with 43 additions and 17 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -917,16 +917,6 @@ register_model_group(
    template="mistral",
 )

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

 register_model_group(
    models={
@@ -1067,6 +1057,18 @@ register_model_group(
 )


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