From 09107affda4d25b7975f5dae34ccc954cd3dc8ea Mon Sep 17 00:00:00 2001
From: sliderSun <291952004@qq.com>
Date: Sun, 7 Apr 2024 10:26:13 +0800
Subject: [PATCH] support Qwen1.5-32B

Former-commit-id: 77044d9ef4ce39a038ad769f93569d5eb2f3bfb0
---
 src/llmtuner/extras/constants.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/llmtuner/extras/constants.py b/src/llmtuner/extras/constants.py
index 6e46218b..19bff11b 100644
--- a/src/llmtuner/extras/constants.py
+++ b/src/llmtuner/extras/constants.py
@@ -663,6 +663,10 @@ register_model_group(
             DownloadSource.DEFAULT: "Qwen/Qwen1.5-14B",
             DownloadSource.MODELSCOPE: "qwen/Qwen1.5-14B",
         },
+        "Qwen1.5-32B": {
+            DownloadSource.DEFAULT: "Qwen/Qwen1.5-32B",
+            DownloadSource.MODELSCOPE: "qwen/Qwen1.5-32B",
+        },
         "Qwen1.5-72B": {
             DownloadSource.DEFAULT: "Qwen/Qwen1.5-72B",
             DownloadSource.MODELSCOPE: "qwen/Qwen1.5-72B",
@@ -691,6 +695,10 @@ register_model_group(
             DownloadSource.DEFAULT: "Qwen/Qwen1.5-14B-Chat",
             DownloadSource.MODELSCOPE: "qwen/Qwen1.5-14B-Chat",
         },
+        "Qwen1.5-32B-Chat": {
+            DownloadSource.DEFAULT: "Qwen/Qwen1.5-32B-Chat",
+            DownloadSource.MODELSCOPE: "qwen/Qwen1.5-32B-Chat",
+        },
         "Qwen1.5-72B-Chat": {
             DownloadSource.DEFAULT: "Qwen/Qwen1.5-72B-Chat",
             DownloadSource.MODELSCOPE: "qwen/Qwen1.5-72B-Chat",