File size: 1,574 Bytes
3c2049d fca5a08 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
---
license: apache-2.0
---
# ChatGLM3 模型量化
本项目用于演示如何使用 ChatGLM3 模型，并通过 ChatGLM.cpp 工具仓库对模型进行量化。
## 拉取模型和工具仓库
首先，需要将 ChatGLM3 模型和 ChatGLM.cpp 工具仓库拉取到本地：
```shell
git lfs install
git clone https://huggingface.co/THUDM/chatglm3-6b
git clone https://github.com/li-plus/chatglm.cpp.git
```
## 安装依赖
在开始量化之前，请确保已安装以下依赖项：
```shell
python3 -m pip install -U -q pip
python3 -m pip install -q torch tabulate tqdm transformers accelerate sentencepiece
```
## 模型量化
使用 ChatGLM.cpp 工具仓库中的 `convert.py` 脚本，可以对 ChatGLM3 模型进行量化。
```shell
types=("q4_0" "q4_1" "q5_0" "q5_1" "q8_0")
for type_str in "${types[@]}"; do
python3 ./chatglm.cpp/chatglm_cpp/convert.py -i ./chatglm3-6b -t ${type_str} -o chatglm-ggml_${type_str}.bin
done
```
上述代码会对 ChatGLM3 模型进行多个量化操作，并生成对应的量化模型文件。量化类型包括："q4_0"、"q4_1"、"q5_0"、"q5_1" 和 "q8_0"。
请确保已在终端中切换到正确的工作目录，然后执行上述代码段。
量化后的模型文件将以 `chatglm-ggml_{type_str}.bin` 的形式保存在当前目录下。
---
如果您有任何问题，请随时联系我们。
模型地址：[https://huggingface.co/THUDM/chatglm3-6b](https://huggingface.co/THUDM/chatglm3-6b)
工具仓库地址：[https://github.com/li-plus/chatglm.cpp](https://github.com/li-plus/chatglm.cpp)
|