From ececd68f9a122a78917435cfd9a5a09a16b8649f Mon Sep 17 00:00:00 2001
From: Kingsley <kingsleydodonow@gmail.com>
Date: Mon, 30 Sep 2024 20:04:47 +0800
Subject: [PATCH] register model fix

Former-commit-id: 15d555c8c523ac2252d55614773073c57db0e025
---
 src/llamafactory/extras/constants.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/llamafactory/extras/constants.py b/src/llamafactory/extras/constants.py
index ed1aff63..4a8a6d25 100644
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -1164,7 +1164,7 @@ register_model_group(
 
 register_model_group(
     models={
-        "Pixtral-12B-Instruct": {
+        "Pixtral-12B-Chat": {
             DownloadSource.DEFAULT: "mistral-community/pixtral-12b",
             DownloadSource.MODELSCOPE: "AI-ModelScope/pixtral-12b",
         }