tiny fix

Former-commit-id: 35bc71b2a68fd303798c35fe22ad29ceea87cf9b
2024-09-28 22:50:53 +08:00
parent e4c57f54f8
commit bddb2646bd
5 changed files with 21 additions and 19 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -1060,8 +1060,8 @@ register_model_group(
 register_model_group(
    models={
        "Pixtral-12B-2409": {
-            DownloadSource.DEFAULT: "mistral-community/pixtral-12b",
-            DownloadSource.MODELSCOPE: "AI-ModelScope/pixtral-12b",
+            DownloadSource.DEFAULT: "mistralai/Pixtral-12B-2409",
+            DownloadSource.MODELSCOPE: "LLM-Research/Pixtral-12B-2409",
        }
    },
    template="mistral",