add TeleChat-1B

2024-07-02 17:49:04 +08:00 · 2024-07-02 17:49:04 +08:00 · 57b7c00430
parent 33f2ddb8b6
commit 57b7c00430
1 changed files with 4 additions and 0 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@ -1260,6 +1260,10 @@ register_model_group(

 register_model_group(
    models={
+        "TeleChat-1B-Chat": {
+            DownloadSource.DEFAULT: "Tele-AI/TeleChat-1B",
+            DownloadSource.MODELSCOPE: "TeleAI/TeleChat-1B",
+        },
        "TeleChat-7B-Chat": {
            DownloadSource.DEFAULT: "Tele-AI/telechat-7B",
            DownloadSource.MODELSCOPE: "TeleAI/telechat-7B",