Tiny fix

2024-09-29 00:00:23 +08:00
parent 3d3cc6705d
commit 8f13a3627d
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",