部署DB-GPT

参考文献

https://zhuanlan.zhihu.com/p/629467580
https://blog.csdn.net/qq_40231723/article/details/133956318

安装环境与模型运行

安装conda环境

首先我们需要安装conda环境,我们使用miniconda3

wget https://repo.anaconda.com/miniconda/Miniconda3-py310_23.3.1-0-Linux-x86_64.sh
sh Miniconda3-py310_23.3.1-0-Linux-x86_64.sh

安装完成之后,需要生效一下环境变量

source /root/.bashrc

conda create -n dbgpt_env python=3.10 
conda activate dbgpt_env

下载git-lfs

下载git-lfs,为了大模型的传输

apt-get install git-lfs

初始化

git lfs install

克隆源码

git clone https://github.com/eosphoros-ai/DB-GPT.git

下载大模型到models目录(huggingface无法访问,使用该镜像网站)

git clone https://www.modelscope.cn/thomas/text2vec-large-chinese.git

git clone https://www.modelscope.cn/ZhipuAI/chatglm2-6b.git

配置.env文件

cd /home/DB-GPT
cp .env.template .env

可以在.env中配置基本参数,例如将LLM_MODEL设置为要使用的模型,LLM_MODEL官方推荐使用vicuna-13b-v1.5。这里我使用的是chatglm2-6b

在这里插入图片描述

加载SQLite的数据

bash ./scripts/examples/load_examples.sh

运行DB-GPT

运行db-gpt服务端

python pilot/server/llmserver.py

服务端运行成功图
在这里插入图片描述

运行客户端

python pilot/server/dbgpt_server.py

客户端运行成功图,端口5000
在这里插入图片描述

浏览器访问
在这里插入图片描述

配置sqlite数据库

点击左侧Data Center,再点击sqlite,修改Path
在这里插入图片描述

数据库示例

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
问题1:用中文提问,模型翻译的单词与数据库列名不一致
在这里插入图片描述
可见模型把学分翻译为score,而不是credit,导致报错
解决思路:涉及到的表名和列名,用英文输入
在这里插入图片描述
问题2:查询总分最高的学生
在这里插入图片描述
在这里插入图片描述

pydantic版本问题

报错信息:pydantic.errors.PydanticUserError: If you use @root_validator with pre=False (the default) you MUST specify skip_on_failure=True. Note that @root_validator is deprecated and should be replaced with @model_validator.

解决:卸载该版本,下载旧版本

pip uninstall pydantic

pip install pydantic==1.10.5

bash报错

在这里插入图片描述
原因:linux系统换行符出现问题
解决:sed -i 's/\r//g' xxx.sh

  • 3
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
要在Docker中部署一个Chatbot GPT-2,您需要按照以下步骤操作: 1. 首先,您需要安装Docker和Docker Compose。您可以在官方网站上找到有关如何安装这些工具的文档。 2. 接下来,您需要下载Chatbot GPT-2代码并将其放置在一个文件夹中。您可以从GitHub或其他来源获取该代码。 3. 在该文件夹中,您需要创建一个名为“ Dockerfile”的文件,并将以下内容复制到该文件中: ``` FROM tensorflow/tensorflow:latest-gpu-py3 RUN apt-get update && apt-get install -y git RUN git clone https://github.com/huggingface/transformers.git && \ cd transformers && git checkout v3.0.2 && \ pip install . RUN git clone https://github.com/microsoft/DialoGPT.git && \ cd DialoGPT && git checkout v1.0 && \ pip install -r requirements.txt WORKDIR /app COPY . . CMD ["python", "main.py"] ``` 这个Dockerfile使用最新的TensorFlow GPU映像,安装了必要的软件包和依赖项,然后将应用程序代码复制到容器中并启动主程序。 4. 接下来,您需要创建一个名为“ docker-compose.yml”的文件,并将以下内容复制到该文件中: ``` version: '3' services: chatbot: build: . expose: - 5000 ports: - 5000:5000 environment: - GPU=1 ``` 这个docker-compose文件定义了一个服务,使用刚才创建的Dockerfile进行构建。它将端口5000暴露给主机,并将容器端口5000映射到主机端口5000。它还设置了一个名为GPU的环境变量,以便容器可以使用GPU。 5. 现在您可以运行以下命令来构建和启动Chatbot GPT-2容器: ``` docker-compose build docker-compose up -d ``` 第一条命令将构建容器,而第二条命令将启动容器并在后台运行。现在,您可以访问http://localhost:5000来与Chatbot GPT-2进行交互。 希望这可以帮助您部署Chatbot GPT-2容器。
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值