From 0f727b393ebfcd7126f858c303172f71c22c20d3 Mon Sep 17 00:00:00 2001
From: hiyouga <hiyouga@buaa.edu.cn>
Date: Sun, 29 Oct 2023 13:30:20 +0800
Subject: [PATCH] update constants

Former-commit-id: ebacbb1072045924a7e335cc9dda488d6f0be8b3
---
 src/llmtuner/extras/constants.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/llmtuner/extras/constants.py b/src/llmtuner/extras/constants.py
index 761dba71..dc55a080 100644
--- a/src/llmtuner/extras/constants.py
+++ b/src/llmtuner/extras/constants.py
@@ -57,6 +57,8 @@ SUPPORTED_MODELS = {
     "XVERSE-13B": "xverse/XVERSE-13B",
     "XVERSE-13B-Chat": "xverse/XVERSE-13B-Chat",
     "ChatGLM2-6B-Chat": "THUDM/chatglm2-6b",
+    "ChatGLM3-6B-Base": "THUDM/chatglm3-6b-base",
+    "ChatGLM3-6B-Chat": "THUDM/chatglm3-6b",
     "Phi1.5-1.3B": "microsoft/phi-1_5"
 }
 
@@ -73,6 +75,7 @@ DEFAULT_MODULE = {
     "Qwen": "c_attn",
     "XVERSE": "q_proj,v_proj",
     "ChatGLM2": "query_key_value",
+    "ChatGLM3": "query_key_value",
     "Phi1.5": "Wqkv"
 }
 
@@ -84,5 +87,6 @@ DEFAULT_TEMPLATE = {
     "InternLM": "intern",
     "Qwen": "chatml",
     "XVERSE": "xverse",
-    "ChatGLM2": "chatglm2"
+    "ChatGLM2": "chatglm2",
+    "ChatGLM3": "chatglm3"
 }