add TeleChat-12B/TeleChat-12B-v2 models

This commit is contained in:
hzhaoy committed 2024-05-29 15:00:37 +08:00
1 parent 4728fe62d1
commit 0dd632fe9e
2 files changed
+26

No files matched your search

+16
View File
@@ -1355,3 +1355,19 @@ register_model_group(
},
template="zephyr",
)
register_model_group(
models={
"TeleChat-12B-Chat": {
DownloadSource.DEFAULT: "Tele-AI/TeleChat-12B",
DownloadSource.MODELSCOPE: "TeleAI/TeleChat-12B",
},
"TeleChat-12B-v2-Chat": {
DownloadSource.DEFAULT: "Tele-AI/TeleChat-12B-v2",
DownloadSource.MODELSCOPE: "TeleAI/TeleChat-12B-v2",
},
},
module='query,key_value',
template="telechat",
)