423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-07-31 10:42:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

[model] update glm4.5 (#8770)

Browse Source
This commit is contained in:
Kingsley 2025-07-29 19:57:29 +08:00 committed by GitHub
parent 3307ff1d4a
commit 4380b7b35e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 41 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/llamafactory/data/template.py
View File

@ -1096,10 +1096,12 @@ register_template(
format_assistant=StringFormatter(slots=["{{content}}<|end_of_text|>\n"]), format_assistant=StringFormatter(slots=["{{content}}<|end_of_text|>\n"]),
format_system=StringFormatter(slots=["<|start_of_role|>system<|end_of_role|>{{content}}<|end_of_text|>\n"]), format_system=StringFormatter(slots=["<|start_of_role|>system<|end_of_role|>{{content}}<|end_of_text|>\n"]),
format_function=FunctionFormatter(slots=["{{content}}<|end_of_text|>\n"], tool_format="default"), format_function=FunctionFormatter(slots=["{{content}}<|end_of_text|>\n"], tool_format="default"),
format_observation=StringFormatter(slots=["<|start_of_role|>tool<|end_of_role|>{{content}}<|end_of_text|>\n<|start_of_role|>assistant\n"]), format_observation=StringFormatter(
slots=["<|start_of_role|>tool<|end_of_role|>{{content}}<|end_of_text|>\n<|start_of_role|>assistant\n"]
),
format_tools=ToolFormatter(tool_format="default"), format_tools=ToolFormatter(tool_format="default"),
stop_words=["<|end_of_text|>"], stop_words=["<|end_of_text|>"],
default_system=("You are Granite, developed by IBM. You are a helpful AI assistant.") default_system=("You are Granite, developed by IBM. You are a helpful AI assistant."),
) )

51
src/llamafactory/extras/constants.py
View File

@ -276,7 +276,7 @@ register_model_group(
register_model_group( register_model_group(
models={ models={
"ChatGLM2-6B-Chat": { "ChatGLM2-6B-Chat": {
DownloadSource.DEFAULT: "THUDM/chatglm2-6b", DownloadSource.DEFAULT: "zai-org/chatglm2-6b",
DownloadSource.MODELSCOPE: "ZhipuAI/chatglm2-6b", DownloadSource.MODELSCOPE: "ZhipuAI/chatglm2-6b",
} }
}, },
@ -287,11 +287,11 @@ register_model_group(
register_model_group( register_model_group(
models={ models={
"ChatGLM3-6B-Base": { "ChatGLM3-6B-Base": {
DownloadSource.DEFAULT: "THUDM/chatglm3-6b-base", DownloadSource.DEFAULT: "zai-org/chatglm3-6b-base",
DownloadSource.MODELSCOPE: "ZhipuAI/chatglm3-6b-base", DownloadSource.MODELSCOPE: "ZhipuAI/chatglm3-6b-base",
}, },
"ChatGLM3-6B-Chat": { "ChatGLM3-6B-Chat": {
DownloadSource.DEFAULT: "THUDM/chatglm3-6b", DownloadSource.DEFAULT: "zai-org/chatglm3-6b",
DownloadSource.MODELSCOPE: "ZhipuAI/chatglm3-6b", DownloadSource.MODELSCOPE: "ZhipuAI/chatglm3-6b",
}, },
}, },
@ -333,7 +333,7 @@ register_model_group(
register_model_group( register_model_group(
models={ models={
"CodeGeeX4-9B-Chat": { "CodeGeeX4-9B-Chat": {
DownloadSource.DEFAULT: "THUDM/codegeex4-all-9b", DownloadSource.DEFAULT: "zai-org/codegeex4-all-9b",
DownloadSource.MODELSCOPE: "ZhipuAI/codegeex4-all-9b", DownloadSource.MODELSCOPE: "ZhipuAI/codegeex4-all-9b",
}, },
}, },
@ -840,28 +840,28 @@ register_model_group(
register_model_group( register_model_group(
models={ models={
"GLM-4-9B": { "GLM-4-9B": {
DownloadSource.DEFAULT: "THUDM/glm-4-9b", DownloadSource.DEFAULT: "zai-org/glm-4-9b",
DownloadSource.MODELSCOPE: "ZhipuAI/glm-4-9b", DownloadSource.MODELSCOPE: "ZhipuAI/glm-4-9b",
}, },
"GLM-4-9B-Chat": { "GLM-4-9B-Chat": {
DownloadSource.DEFAULT: "THUDM/glm-4-9b-chat", DownloadSource.DEFAULT: "zai-org/glm-4-9b-chat",
DownloadSource.MODELSCOPE: "ZhipuAI/glm-4-9b-chat", DownloadSource.MODELSCOPE: "ZhipuAI/glm-4-9b-chat",
DownloadSource.OPENMIND: "LlamaFactory/glm-4-9b-chat", DownloadSource.OPENMIND: "LlamaFactory/glm-4-9b-chat",
}, },
"GLM-4-9B-1M-Chat": { "GLM-4-9B-1M-Chat": {
DownloadSource.DEFAULT: "THUDM/glm-4-9b-chat-1m", DownloadSource.DEFAULT: "zai-org/glm-4-9b-chat-1m",
DownloadSource.MODELSCOPE: "ZhipuAI/glm-4-9b-chat-1m", DownloadSource.MODELSCOPE: "ZhipuAI/glm-4-9b-chat-1m",
}, },
"GLM-4-0414-9B-Chat": { "GLM-4-0414-9B-Chat": {
DownloadSource.DEFAULT: "THUDM/GLM-4-9B-0414", DownloadSource.DEFAULT: "zai-org/GLM-4-9B-0414",
DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4-9B-0414", DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4-9B-0414",
}, },
"GLM-4-0414-32B-Base": { "GLM-4-0414-32B-Base": {
DownloadSource.DEFAULT: "THUDM/GLM-4-32B-Base-0414", DownloadSource.DEFAULT: "zai-org/GLM-4-32B-Base-0414",
DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4-32B-Base-0414", DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4-32B-Base-0414",
}, },
"GLM-4-0414-32B-Chat": { "GLM-4-0414-32B-Chat": {
DownloadSource.DEFAULT: "THUDM/GLM-4-32B-0414", DownloadSource.DEFAULT: "zai-org/GLM-4-32B-0414",
DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4-32B-0414", DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4-32B-0414",
}, },
}, },
@ -872,11 +872,11 @@ register_model_group(
register_model_group( register_model_group(
models={ models={
"GLM-4.1V-9B-Base": { "GLM-4.1V-9B-Base": {
DownloadSource.DEFAULT: "THUDM/GLM-4.1V-9B-Base", DownloadSource.DEFAULT: "zai-org/GLM-4.1V-9B-Base",
DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4.1V-9B-Base", DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4.1V-9B-Base",
}, },
"GLM-4.1V-9B-Thinking": { "GLM-4.1V-9B-Thinking": {
DownloadSource.DEFAULT: "THUDM/GLM-4.1V-9B-Thinking", DownloadSource.DEFAULT: "zai-org/GLM-4.1V-9B-Thinking",
DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4.1V-9B-Thinking", DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4.1V-9B-Thinking",
}, },
}, },
@ -885,14 +885,37 @@ register_model_group(
) )
register_model_group(
models={
"GLM-4.5-Air-Base": {
DownloadSource.DEFAULT: "zai-org/GLM-4.5-Air-Base",
DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4.5-Air-Base",
},
"GLM-4.5-Base": {
DownloadSource.DEFAULT: "zai-org/GLM-4.5-Base",
DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4.5-Base",
},
"GLM-4.5-Air-Chat": {
DownloadSource.DEFAULT: "zai-org/GLM-4.5-Air",
DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4.5-Air",
},
"GLM-4.5-Chat": {
DownloadSource.DEFAULT: "zai-org/GLM-4.5",
DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4.5",
},
},
template="glm4_moe",
)
register_model_group( register_model_group(
models={ models={
"GLM-Z1-0414-9B-Chat": { "GLM-Z1-0414-9B-Chat": {
DownloadSource.DEFAULT: "THUDM/GLM-Z1-9B-0414", DownloadSource.DEFAULT: "zai-org/GLM-Z1-9B-0414",
DownloadSource.MODELSCOPE: "ZhipuAI/GLM-Z1-9B-0414", DownloadSource.MODELSCOPE: "ZhipuAI/GLM-Z1-9B-0414",
}, },
"GLM-Z1-0414-32B-Chat": { "GLM-Z1-0414-32B-Chat": {
DownloadSource.DEFAULT: "THUDM/GLM-Z1-32B-0414", DownloadSource.DEFAULT: "zai-org/GLM-Z1-32B-0414",
DownloadSource.MODELSCOPE: "ZhipuAI/GLM-Z1-32B-0414", DownloadSource.MODELSCOPE: "ZhipuAI/GLM-Z1-32B-0414",
}, },
}, },