京东质量社区 ChatGPT 小型平替之 ChatGLM-6B 本地化部署、接入本地知识库体验 | 京东云技术团队

京东云开发者 · 2023年06月20日 · 最后由 liudapeng 回复于 2023年06月26日 · 6476 次阅读

本文期望通过本地化部署一个基于 LLM 模型的应用,能让大家对构建一个完整的应用有一个基本认知。包括基本的软硬环境依赖、底层的 LLM 模型、中间的基础框架及最上层的展示组件,最终能达到在本地零编码体验的目的。

一、ChatGLM-6B 模型介绍

https://github.com/THUDM/ChatGLM-6B [Star 27.6k]

一个清华开源的、支持中英双语的对话语言模型,基于 GLM 架构,62 亿参数。可以本地安装部署运行在消费级的显卡上做模型的推理和训练。

  • 开源 10 天 10000stars

  • 当天在 GitHub 的趋势排行第一

  • Huggingface 下载超过 100 万

  • 开源的训练数据量达到 1 万亿字符的模型

1、能力

  • 自我认知

  • 文案写作

  • 提纲写作

  • 信息抽取

2、缺点

  • 模型容量小导致回答存在偏见内容,推理能力较弱

3、场景

  • 垂直领域知识

  • 基于私有数据的问答

二、部署体验

1、环境依赖:

硬件:

  • 基于 GPU,建议 16GB 起步,建议 24GB+ 体验,否则多轮容易爆显存;

  • 无 GPU 可以仅用 CPU,大约需要 25GB 内存,CPU 会慢一点,本次不使用。

软件:

  • CUDA 11.7+

  • Python3.10.8+

  • pip3

  • git

2、钞能力 - 算力市场:

强烈不建议本地部署,一整套硬件价值不菲,费时耗力,按需购买算力对于体验来说最划算,建议使用京东 Ku+ 平台即可,目前都有配额,可直接申请使用,本次演示流程基于 autodl 算力平台搭建:

▪http://kuplus.jd.com [Ku+]

▪https://www.autodl.com [AutoDL]

3、下载 demo:

# 下载项目源代码
git clone https://github.com/THUDM/ChatGLM-6B
# 切换到项目根目录
cd ChatGLM-6B
# 安装依赖
pip install -r requirements.txt
# 安装回老版本gradio,解决输出内容未渲染html标签问题
pip install gradio==3.28.3

# 如果pip下载有问题,修改其他源,比例douban或aliyun,没有pip.conf需新建
mkdir ~/.pip && touch ~/.pip/pip.conf && vim ~/.pip/pip.conf

[global]
index-url=http://pypi.douban.com/simple/
[install] 
trusted-host=pypi.douban.com

4、下载模型:

# 直接git下载模型,大约需要13G空间
git clone https://huggingface.co/THUDM/chatglm-6b

# 或者部分地区可能无法下载,可以使用镜像单独下载,新建url.txt, 拷贝以下镜像地址保存
https://cloud.tsinghua.edu.cn/seafhttp/files/08ff8050-912c-47b8-ad5c-56bcd231df71/ice_text.model
https://cloud.tsinghua.edu.cn/seafhttp/files/88a7978c-8eda-498d-85e8-0671294a4c47/pytorch_model-00001-of-00008.bin
https://cloud.tsinghua.edu.cn/seafhttp/files/de998e41-7093-41b3-84f1-59e32361e703/pytorch_model-00002-of-00008.bin
https://cloud.tsinghua.edu.cn/seafhttp/files/077b5058-5ebd-4930-9470-8e873c01f47c/pytorch_model-00003-of-00008.bin
https://cloud.tsinghua.edu.cn/seafhttp/files/728f7324-ce96-44bc-a08f-5c7222727ca5/pytorch_model-00004-of-00008.bin
https://cloud.tsinghua.edu.cn/seafhttp/files/932b40ac-d195-4ba4-8d06-7946b8e6d0d8/pytorch_model-00005-of-00008.bin
https://cloud.tsinghua.edu.cn/seafhttp/files/d763678e-c438-416b-a4b9-5594c52cb3f6/pytorch_model-00006-of-00008.bin
https://cloud.tsinghua.edu.cn/seafhttp/files/59da96fd-8eb9-42d4-992e-03e1d0637a01/pytorch_model-00007-of-00008.bin
https://cloud.tsinghua.edu.cn/seafhttp/files/849d7de4-9253-487a-bc37-9a43418e3c0c/pytorch_model-00008-of-00008.bin
# 此处文件夹名称使用THUDM/chatglm-6b
mkdir -p THUDM/chatglm-6b && cd THUDM/chatglm-6b && touch url.txt
# 保存批量下载地址
vim url.txt
# 执行wget批量下载到当前文件夹
wget -i url.txt

5、安装运行:

打开 web_demo.py

# 一、如果模型位置不是默认path,修改如下位置:
tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True)
model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).half().cuda()

# 二、找到最后一行或者类似 queue().launch 位置,指定ip、端口方便外网映射用,一般算力市场会默认提供一个映射端口,指定即可:
demo.queue().launch(share=False, inbrowser=True, server_name='0.0.0.0', server_port=6006)

# 启动
python web_demo.py

root@autodl-container-9494499a62-3e5ab6d1:~/ChatGLM-6B# python web_demo.py 
Explicitly passing a `revision` is encouraged when loading a model with custom code to ensure no malicious code has been contributed in a newer revision.
Explicitly passing a `revision` is encouraged when loading a configuration with custom code to ensure no malicious code has been contributed in a newer revision.
Explicitly passing a `revision` is encouraged when loading a model with custom code to ensure no malicious code has been contributed in a newer revision.
Loading checkpoint shards: 100%|█████████████████████████████████████████████████████████████████████| 8/8 [00:10<00:00,  1.36s/it]
Running on local URL:  http://0.0.0.0:6006

To create a public link, set `share=True` in `launch()`.

6、实际效果、资源占用:

主机配置 演示效果 
 点击查看大图 

7、推理参数含义:

· Maximum length

通常用于限制输入序列的最大长度,因为 ChatGLM-6B 是 2048 长度推理的,一般这个保持默认就行, 太大可能会导致性能下降

· Top P

Top P 参数是指在生成文本等任务中,选择可能性最高的前 P 个词的概率累加和。这个参数被称为 Top P,也称为 Nucleus Sampling。

例如,如果将 Top P 参数设置为 0.7,那么模型会选择可能性排名超过 70% 的词进行采样。这样可以保证生成的文本准确性较高,但可能会缺乏多样性。相反,如果将 Top P 参数设置为 0.3,则会选择可能性超过 30% 的词进行采样,这可能会导致生成文本的准确性下降,但能够更好地增加多样性。

· Temperature

Temperature 参数通常用于调整 softmax 函数的输出,用于增加或减少模型对不同类别的置信度。

具体来说,softmax 函数将模型对每个类别的预测转换为概率分布。Temperature 参数可以看作是一个缩放因子,它可以增加或减少 softmax 函数输出中每个类别的置信度。

例如,将 Temperature 设置为 0.05 和 0.95 的主要区别在于,T=0.05 会使得模型更加自信,更加倾向于选择概率最大的类别作为输出,而 T=0.95 会使得模型更加不确定,更加倾向于输出多个类别的概率值较大。

8、其他好玩的:

看图说话,VisualGLM-6B,一个支持图像理解的多模态对话语言模型,语言模型基于 ChatGLM-6B。

  

三、结合 LangChain 实现本地知识库

1、LangChain:

https://github.com/hwchase17/langchain [Star 45k]

⽤于开发由语⾔模型驱动的应⽤程序的基础框架,是一个链接面向用户程序和 LLM 之间的中间层,可以理解成是一个 lib。

· 简介原理

· 主要功能

1.调⽤语⾔模型;

2.将不同数据源接⼊到语⾔模型的交互中;

3.允许语⾔模型与运⾏环境交互。

· 模块介绍

1.Modules:⽀持的模型类型和集成;

2.Prompt:提示词管理、优化和序列化;

3.Memory:内存是指在链/代理调⽤之间持续存在的状态;

4.Indexes:当语⾔模型与特定于应⽤程序的数据相结合时,会变得更加强⼤ - 此模块包含⽤于加载、查询和更新外部数据的接⼝和集成;

5.Chain:链是结构化的调⽤序列(对 LLM 或其他实⽤程序);

6.Agents:代理是⼀个链,其中 LLM 在给定⾼级指令和⼀组⼯具的情况下,反复决定操作,执⾏操作并观察结果,直到⾼级指令执行完成;

7.Callbacks:回调允许您记录和流式传输任何链的中间步骤,从⽽轻松观察、调试和评估应⽤程序的内部。

2、LangChain-ChatGLM

https://github.com/imClumsyPanda/langchain-ChatGLM [Star 8k]

一种利用 langchain 实现的基于本地知识库的问答应用,目标期望建立一套对中文场景与开源模型支持友好、可离线运行的知识库问答解决方案。

实现过程:包括加载文件 -> 读取文本 -> 文本分割 -> 文本向量化 -> 问句向量化 -> 在文本向量中匹配出与问句向量最相似的 top k 个 -> 匹配出的文本作为上下文和问题一起添加到 prompt 中 -> 提交给 LLM 生成回答。

· 项⽬特点

1.依托 ChatGLM 等开源模型实现,可离线部署;

2.基于 langchain 实现,可快速实现接⼊多种数据源;

3.在分句、⽂档读取等⽅⾯,针对中⽂使⽤场景优化;

4.⽀持 pdf、txt、md、docx 等⽂件类型接⼊,具备命令⾏ demo、webui 和 vue 前端。

· 项⽬结构

1.models:llm 的接⼝类与实现类,针对开源模型提供流式输出⽀持;

2.loader:⽂档加载器的实现类;

3.textsplitter:⽂本切分的实现类;

4.chains:⼯作链路实现,如 chains/local_doc_qa 实现了基于本地⽂档的问答实现;

5.content:⽤于存储上传的原始⽂件;

6.vector_store:⽤于存储向量库⽂件,即本地知识库本体;

7.configs:配置⽂件存储。

· 安装部署

与 ChatGLM 安装基本类似,git clone 之后修改模型地址到本地 chartglm-6b 的 path 启动即可,如果使用 AutoDL 平台,可用镜像一键拉起,镜像保持最新 master 版本,可体验最新功能。

最新镜像地址,已包含离线 llm model: chatglm-6b 和 embedding: text2vec-large-chinese: https://www.codewithgpu.com/i/Liudapeng/langchain-ChatGLM/langchain-chatglm-master

# 新增一步,下载离线 embedding model 
sudo apt-get install git-lfs
git lfs install
git clone https://huggingface.co/GanymedeNil/text2vec-large-chinese

# 启用本地离线模型
vim ~/langchain-ChatGLM/configs/model_config.py

# 1、修改 embedding path
embedding_model_dict = {
    "text2vec": "/root/GanymedeNil/text2vec-large-chinese",
}

# 2、修改 llm path, 直接指向为第一章节的 chatglm-6b path即可
"chatglm-6b": {
    "name": "chatglm-6b",
    "pretrained_model_name": "THUDM/chatglm-6b",
    "local_model_path": "/root/ChatGLM-6B/THUDM/chatglm-6b",
    "provides": "ChatGLM"
}

# 3、启用本地模型
NO_REMOTE_MODEL = True

# 运行Gradio web demo
cd ~/langchain-ChatGLM/

root@autodl-container-95b111aeb0-c6fdac9f:~/langchain-ChatGLM# python webui.py 
INFO  2023-06-08 14:10:58,531-1d: 
loading model config
llm device: cuda
embedding device: cuda
dir: /root/langchain-ChatGLM
flagging username: dc706ff885da43a8b0e8181bcc72ad1e

WARNING: OMP_NUM_THREADS set to 14, not 1. The computation speed will not be optimized if you use data parallel. It will fail if this PaddlePaddle binary is 
compiled with OpenBlas since OpenBlas does not support multi-threads.
PLEASE USE OMP_NUM_THREADS WISELY.
Loading chatglm-6b...
Loading checkpoint shards: 100%|██████████████████████████████████████████████████████████████████████████████| 8/8 [00:13<00:00,  1.66s/it]
Loaded the model in 23.48 seconds.
INFO  2023-06-08 14:11:26,780-1d: Load pretrained SentenceTransformer: /root/GanymedeNil/text2vec-large-chinese
{'answer': '你好👋!我是人工智能助手 ChatGLM-6B,很高兴见到你,欢迎问我任何问题。'}
INFO  2023-06-08 14:11:33,694-1d: 模型已成功加载,可以开始对话,或从右侧选择模式后开始对话
Running on local URL:  http://0.0.0.0:6006

To create a public link, set `share=True` in `launch()`.
· 效果对比
LLM 回答 LLM+ 知识库 知识库来源
  

四、视图框架

快速构建针对人工智能的 python 的 webApp 库,封装前端页面 + 后端接口 +AI 算法模型推理,方便 AI 算法工程师快速展示成果,常用的两个展示层框架:

1、Gradio:

优势在于易用性,代码结构相比 Streamlit 简单,只需简单定义输入和输出接口即可快速构建简单的交互页面,更轻松部署模型。适合场景相对简单,想要快速部署应用的开发者;

2、Streamlit:

优势在于可扩展性,相比 Gradio 复杂。适合场景相对复杂,想要构建丰富多样交互页面的开发者。

作者:京东科技 刘大朋

来源:京东云开发者社区

共收到 6 条回复 时间 点赞

创建完怎么访问

训练集从哪来啊,特别是多模态这块的,还是这个开源项目已经全部包含了

3楼 已删除
Atlantis 回复

安装部署环节有说怎么修改 ip 和端口号,直接设置成算力市场的端口号就行了,算力平台会提供一个外网访问地址,ip+ 端口访问就行了。

5楼 已删除
匿名用户 回复

整个不涉及训练,只是个应用实践而已,用的清华已经训练好的模型,也不建议自己训练哈。

需要 登录 后方可回复, 如果你还没有账号请点击这里 注册