diff --git a/README.md b/README.md index b836487e..0b486c1d 100644 --- a/README.md +++ b/README.md @@ -239,7 +239,7 @@ Compared to ChatGLM's [P-Tuning](https://github.com/THUDM/ChatGLM2-6B/tree/main/ | [Falcon](https://huggingface.co/tiiuae) | 7B/11B/40B/180B | falcon | | [Gemma/Gemma 2/CodeGemma](https://huggingface.co/google) | 2B/7B/9B/27B | gemma | | [Gemma 3](https://huggingface.co/google) | 1B/4B/12B/27B | gemma3/gemma (1B) | -| [GLM-4](https://huggingface.co/THUDM) | 9B | glm4 | +| [GLM-4/GLM-4-0414](https://huggingface.co/THUDM) | 9B/32B | glm4 | | [GPT-2](https://huggingface.co/openai-community) | 0.1B/0.4B/0.8B/1.5B | - | | [Granite 3.0-3.1](https://huggingface.co/ibm-granite) | 1B/2B/3B/8B | granite3 | | [Index](https://huggingface.co/IndexTeam) | 1.9B | index | diff --git a/README_zh.md b/README_zh.md index bec4812e..9e520e74 100644 --- a/README_zh.md +++ b/README_zh.md @@ -242,7 +242,7 @@ https://github.com/user-attachments/assets/43b700c6-a178-41db-b1f8-8190a5d3fcfc | [Falcon](https://huggingface.co/tiiuae) | 7B/11B/40B/180B | falcon | | [Gemma/Gemma 2/CodeGemma](https://huggingface.co/google) | 2B/7B/9B/27B | gemma | | [Gemma 3](https://huggingface.co/google) | 1B/4B/12B/27B | gemma3/gemma (1B) | -| [GLM-4](https://huggingface.co/THUDM) | 9B | glm4 | +| [GLM-4/GLM-4-0414](https://huggingface.co/THUDM) | 9B/32B | glm4 | | [GPT-2](https://huggingface.co/openai-community) | 0.1B/0.4B/0.8B/1.5B | - | | [Granite 3.0-3.1](https://huggingface.co/ibm-granite) | 1B/2B/3B/8B | granite3 | | [Hunyuan](https://huggingface.co/tencent/) | 7B | hunyuan | diff --git a/src/llamafactory/extras/constants.py b/src/llamafactory/extras/constants.py index 42a679c1..1c688160 100644 --- a/src/llamafactory/extras/constants.py +++ b/src/llamafactory/extras/constants.py @@ -724,6 +724,26 @@ register_model_group( DownloadSource.DEFAULT: "THUDM/glm-4-9b-chat-1m", DownloadSource.MODELSCOPE: "ZhipuAI/glm-4-9b-chat-1m", }, + "GLM-4-9B-Chat-0414": { + DownloadSource.DEFAULT: "THUDM/GLM-4-9B-Chat-0414", + DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4-9B-Chat-0414" , + }, + "GLM-4-32B-0414": { + DownloadSource.DEFAULT: "THUDM/GLM-4-32B-0414", + DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4-32B-0414" , + }, + "GLM-4-32B-Chat-0414": { + DownloadSource.DEFAULT: "THUDM/GLM-4-32B-Chat-0414", + DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4-32B-Chat-0414" , + }, + "GLM-4-Z1-9B-0414": { + DownloadSource.DEFAULT: "THUDM/GLM-4-Z1-9B-0414", + DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4-Z1-9B-0414" , + }, + "GLM-4-Z1-32B-0414": { + DownloadSource.DEFAULT: "THUDM/GLM-4-Z1-32B-0414", + DownloadSource.MODELSCOPE: "ZhipuAI/GLM-4-Z1-32B-0414" , + }, }, template="glm4", )