add yuan model

This commit is contained in:
hiyouga committed 2023-12-29 13:50:24 +08:00
1 parent 47da742fc9
commit c7ea17d616
3 files changed
+23 -2

No files matched your search

+19
View File
@@ -641,6 +641,25 @@ register_model_group(
)
register_model_group(
models={
"Yuan2-2B-Chat": {
DownloadSource.DEFAULT: "IEITYuan/Yuan2-2B-hf",
DownloadSource.MODELSCOPE: "YuanLLM/Yuan2.0-2B-hf"
},
"Yuan2-51B-Chat": {
DownloadSource.DEFAULT: "IEITYuan/Yuan2-51B-hf",
DownloadSource.MODELSCOPE: "YuanLLM/Yuan2.0-51B-hf"
},
"Yuan2-102B-Chat": {
DownloadSource.DEFAULT: "IEITYuan/Yuan2-102B-hf",
DownloadSource.MODELSCOPE: "YuanLLM/Yuan2.0-102B-hf"
}
},
template="yuan"
)
register_model_group(
models={
"Zephyr-7B-Alpha-Chat": {