Tiny fix

Former-commit-id: ae66e1a545f4cd209a57fd824f9bfb7e94436cba
2024-09-29 00:00:23 +08:00
parent bddb2646bd
commit e641f1215a
5 changed files with 6 additions and 9 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -1059,9 +1059,9 @@ register_model_group(

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