From 1df3f02aca27bc2c970b69e40c8e8c336e7f7e6d Mon Sep 17 00:00:00 2001
From: hzhaoy <hzywong@gmail.com>
Date: Tue, 2 Jul 2024 17:49:04 +0800
Subject: [PATCH] add TeleChat-1B

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

diff --git a/src/llamafactory/extras/constants.py b/src/llamafactory/extras/constants.py
index 3ea21f76..a00411c3 100644
--- 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",