From a32c3a50fcf1eb5e5d1f0f02fbd363548273f2fa Mon Sep 17 00:00:00 2001
From: hiyouga <467089858@qq.com>
Date: Fri, 17 May 2024 13:25:36 +0800
Subject: [PATCH] add deepseek v2 lite model

Former-commit-id: 5e864e6b721d8b891b1cc2ca2dcac41babb9eaaf
---
 src/llamafactory/extras/constants.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/llamafactory/extras/constants.py b/src/llamafactory/extras/constants.py
index 40826163..6b967517 100644
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -323,6 +323,9 @@ register_model_group(
             DownloadSource.DEFAULT: "deepseek-ai/deepseek-moe-16b-base",
             DownloadSource.MODELSCOPE: "deepseek-ai/deepseek-moe-16b-base",
         },
+        "DeepSeek-MoE-16B-v2-Base": {
+            DownloadSource.DEFAULT: "deepseek-ai/DeepSeek-V2-Lite",
+        },
         "DeepSeek-MoE-236B-Base": {
             DownloadSource.DEFAULT: "deepseek-ai/DeepSeek-V2",
             DownloadSource.MODELSCOPE: "deepseek-ai/DeepSeek-V2",
@@ -331,6 +334,9 @@ register_model_group(
             DownloadSource.DEFAULT: "deepseek-ai/deepseek-moe-16b-chat",
             DownloadSource.MODELSCOPE: "deepseek-ai/deepseek-moe-16b-chat",
         },
+        "DeepSeek-MoE-16B-v2-Chat": {
+            DownloadSource.DEFAULT: "deepseek-ai/DeepSeek-V2-Lite-Chat",
+        },
         "DeepSeek-MoE-236B-Chat": {
             DownloadSource.DEFAULT: "deepseek-ai/DeepSeek-V2-Chat",
             DownloadSource.MODELSCOPE: "deepseek-ai/DeepSeek-V2-Chat",