treehouse group(hk)limited
用户1600
添加快捷方式
分享
腾讯云部署清华大学 ChatGLM-6B 模型实战
输入“/”快速插入内容
腾讯云部署清华大学 ChatGLM-6B 模型实战
简介(来自官方)
ChatGLM-6B 是一个开源的、支持中英双语的对话语言模型,基于
General Language Model (GLM)
架构,具有 62 亿参数。结合模型量化技术,用户可以在消费级的
显卡
上进行本地部署(INT4 量化级别下最低只需 6GB 显存)。 ChatGLM-6B 使用了和 ChatGPT 相似的技术,针对中文问答和对话进行了优化。经过约 1T 标识符的中英双语训练,辅以监督微调、反馈自助、人类反馈
强化学习
等技术的加持,62 亿参数的 ChatGLM-6B 已经能生成相当符合人类偏好的回答,更多信息请参考我们的
博客
。
为了方便下游开发者针对自己的应用场景定制模型,我们同时实现了基于
P-Tuning v2
的高效参数微调方法
(使用指南)
,INT4 量化级别下最低只需 7GB 显存即可启动微调。
不过,由于 ChatGLM-6B 的规模较小,目前已知其具有相当多的
局限性
,如事实性/数学逻辑错误,可能生成有害/有偏见内容,较弱的上下文能力,自我认知混乱,以及对英文指示生成与中文指示完全矛盾的内容。请大家在使用前了解这些问题,以免产生误解。更大的基于 1300 亿参数
GLM-130B
的 ChatGLM 正在内测开发中。
准备工作
硬件要求(腾讯云V100
显卡
,一个月大几千,肉疼)
安装git什么的,python我就不说了,太简单
代码clone(怎么安装Git工具不赘述)
代码块
Plain Text
git
clone
https:
//github.com/THUDM/ChatGLM-6B.git
环境安装
使用 pip 安装依赖,其中 transformers 库版本为 4.27.1(就用这个就行)。
代码块
Plain Text
pip install -r
requirements.txt
,
注意:Python要3.8以上!!!!否则能坑死你!
模型以及配置文件准备
1.
官方下载:
清华大学云盘
2.
huggingface开源社区下载:
THUDM/chatglm-6b at main
把对应的模型以及配置文件记得下载全,结果是这样的:
踩坑:记得校验一下模型的SHA256是否跟官方一致,否则模型肯定加载失败!
3.
准备启动:
三种方式均可:
代码块
Plain Text
python3.8 web_demo.py
python3.8 cli_demo.py
streamlit run
web_demo2.py
注意:web方式记得把
云服务器
的端口开放了,否则用不了!
效果展示