add deepseek moe 236B

Former-commit-id: 30c10e2dc41b5d64191a91ad2d61f3b5c440b1d5
This commit is contained in:
hiyouga
2024-05-08 16:37:54 +08:00
parent 6d9d8b92ca
commit 0a1b6ca5a7
5 changed files with 12 additions and 4 deletions

View File

@@ -324,6 +324,14 @@ register_model_group(
DownloadSource.DEFAULT: "deepseek-ai/deepseek-moe-16b-chat",
DownloadSource.MODELSCOPE: "deepseek-ai/deepseek-moe-16b-chat",
},
"DeepSeek-MoE-236B": {
DownloadSource.DEFAULT: "deepseek-ai/DeepSeek-V2",
DownloadSource.MODELSCOPE: "deepseek-ai/DeepSeek-V2",
},
"DeepSeek-MoE-236B-Chat": {
DownloadSource.DEFAULT: "deepseek-ai/DeepSeek-V2-Chat",
DownloadSource.MODELSCOPE: "deepseek-ai/DeepSeek-V2-Chat",
},
},
template="deepseek",
)