知识库chatgpt 轻松搭建本地知识库的ChatGLM2-6B

默认分类3个月前发布 admin
3,502 0
ChatGPT国内版

近期发现了一个项目,它的前身是,在我之前的博客中有关于的部署过程,本项目在前者基础上进行了优化,可以基于当前主流的LLM模型和庞大的知识库,实现本地部署自己的,并可结合自己的知识对模型进行微调,让问答中包含自己上传的知识。依托于本项目支持的开源 LLM 与 模型,本项目可实现全部使用开源模型离线私有部署。与此同时,本项目也支持 GPT API 的调用,并将在后续持续扩充对各类模型及模型 API 的接入。下面具体讲解如何来实现本地部署,过程十分简单,使用起来十分方便。

目录

一、代码下载

项目链接 – -space/-

将该代码下载到本地。

二、模型支持

本项目中默认使用的 LLM 模型为THUDM/-6b,默认使用的 模型为moka-ai/m3e-base为例。

2.1 LLM 模型支持

本项目最新版本中基于进行本地 LLM 模型接入,支持模型如下:

以上模型支持列表可能随更新而持续更新,可参考 已支持模型列表。

除本地模型外,本项目也支持直接接入 API,具体设置可参考/.py.中的的–3.5配置信息。

2.2 模型支持

本项目支持调用中的 模型,已支持的 模型如下:

三、开发部署 3.1 软件需求

本项目已在 3.8.1 – 3.10,CUDA 11.7 环境下完成测试。已在 、ARM 架构的 macOS、Linux 系统中完成测试。

3.2 开发环境准备

参见开发环境准备。

请注意:0.2.3及更新版本的依赖包与0.1.x版本依赖包可能发生冲突,强烈建议新建环境后重新安装依赖包。

首先要创建虚拟环境,然后在虚拟环境中切换到代码文件目录:

cd /d E:DemoLangchain-Chatchat-master   # 切换到代码所在位置

创建虚拟环境:

conda create -n Chat_GLM python=3.9  # 虚拟环境名称为ChatGLM

启用虚拟环境:

conda activate ChatGLM

安装环境依赖:

pip install requirements.txt

安装语言包:

spacy/ at main (.co)

知识库chatgpt 轻松搭建本地知识库的ChatGLM2-6B

只需下载下方图片目录中的whl文件,然后切换到-any-none-any.whl目录,采用pip 安装

pip install zh_core_web_sm-any-none-any.whl

3.3 下载模型至本地

如需在本地或离线环境下运行本项目,需要首先将项目所需的模型下载至本地,通常开源 LLM 与 模型可以从下载。

以本项目中默认使用的 LLM 模型THUDM/-6b与 模型moka-ai/m3e-base为例。直接点击下载。

网盘下载:

链接:

提取码:eaxq

链接:

提取码:0arh

3.4 设置配置项

复制模型相关参数配置模板文件/.py.存储至项目路径下./路径下,并重命名为.py。

复制服务相关参数配置模板文件/.py.存储至项目路径下./路径下,并重命名为.py。

在开始执行 Web UI 或命令行交互前,请先检查/.py和/.py中的各项模型参数设计是否符合需求:

llm_model_dict={
                "chatglm2-6b": {
                        "local_model_path": "E:Demochatglm2-6b",   # 只需该这里
                        "api_base_url": "http://localhost:8888/v1",  # "name"修改为 FastChat 服务中的"api_base_url"
                        "api_key": "EMPTY"
                    },
                }

embedding_model_dict = {
                        "m3e-base": "E:Demom3e-base",   # 只需改这里
                       }

如果你选择使用的模型,请将模型的key写入中。使用该模型,你需要能够访问官的API,或设置代理。

3.5知识库初始化与迁移

当前项目的知识库信息存储在数据库中,在正式运行项目之前请先初始化数据库(我们强烈建议您在执行操作前备份您的知识文件)。

3.6 一键启动API 服务或 Web UI

1. 一键启用默认模型

一键启动脚本 .py,一键启动所有 服务、API 服务、WebUI 服务,示例代码:

python startup.py -a

自动跳转到web页面,若不能跳转记得在代码末尾按一下Enter键,即可一键启动web页面。

并可使用Ctrl + C直接关闭所有运行服务。如果一次结束不了,可以多按几次。

知识库chatgpt 轻松搭建本地知识库的ChatGLM2-6B

可选参数包括-a (或–all-webui),–all-api,–llm-api,-c (或–),—api,-m (或–model-),–api,–webui,其中:

--all-webui 为一键启动 WebUI 所有依赖服务;
--all-api 为一键启动 API 所有依赖服务;
--llm-api 为一键启动 Fastchat 所有依赖的 LLM 服务;
--openai-api 为仅启动 FastChat 的 controller 和 openai-api-server 服务;
其他为单独服务启动选项。

2. 启用非默认模型

若想指定非默认模型,需要用–model-name选项,示例:

python startup.py --all-webui --model-name Qwen-7B-Chat

更多信息可通过 .py -h查看。

四、添加个人知识库

如果当前回答结果并不理想,如下图所示,可添加知识库,对知识进行优化

4.1 本地添加知识库

将文档添加到————目录下:

4.2 通过web页面添加知识库

4.3 重新微调模型

知识库添加完成之后,按照3.5-3.6重新运行程序,即可实现本地知识库更新。如下图所示,添加完《天龙八部》电子书之后,回答结果变得精准。

提升回答速度,把CPU改为GPU:

.py文件中,修改

# Embedding 模型运行设备。设为"auto"会自动检测,也可手动设定为"cuda","mps","cpu"其中之一。
EMBEDDING_DEVICE = "auto"

为:

# Embedding 模型运行设备。设为"auto"会自动检测,也可手动设定为"cuda","mps","cpu"其中之一。
EMBEDDING_DEVICE = "cuda"

修改

# LLM 运行设备。设为"auto"会自动检测,也可手动设定为"cuda","mps","cpu"其中之一。
LLM_DEVICE = "auto"

为:

# LLM 运行设备。设为"auto"会自动检测,也可手动设定为"cuda","mps","cpu"其中之一。
LLM_DEVICE = "cuda"

提示:

若上述部署中遇到问题,可关注我dy:

323AI导航网发布

© 版权声明
广告也精彩

相关文章

暂无评论

暂无评论...