当前位置:   article > 正文

api-for-open-llm对接CodeGeex2部署流程

api-for-open-llm对接CodeGeex2部署流程

使用api-for-open-llm对接CodeGeex2,提供OpenAI接口,部署流程

流程记录

1、环境

阿里云GPU主机,GPU是A10、24G显存

2、涉及到的资源

1、CodeGeex2,参数文件,可以在国内镜像网站下载
2、api-for-open-llm,https://github.com/xusenlinzy/api-for-open-llm

3、本地部署流程

建立conda环境
1、 conda create -n openAI python=3.10
2、 conda activate openAI
clone项目、安装依赖,阿里云里的cuda版本是11.8,安装的torch要适配
3、 git clone https://github.com/xusenlinzy/api-for-open-llm.git
4、 pip install torch2.0.0 torchvision0.15.1 torchaudio2.0.1 --index-url https://download.pytorch.org/whl/cu118
5、 pip install -r requirements.txt
6、 进入目录,cp .env.example .env,
我只修改两项
MODEL_NAME=codegeex2-6b
MODEL_PATH=/mnt/workspace/model/codegeex2-6b
7、启动
cp api/server.py .
python server.py
报错:AttributeError: ‘ChatGLMTokenizer’ object has no attribute ‘tokenizer’. Did you mean: ‘tokenize’?
降低 transformers 版本就可以跑起来了
pip uninstall transformers
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple transformers
4.33.2

8、测试,http://ip:8000/v1/models (我做的内网穿透,ip是代理地址)

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/菜鸟追梦旅行/article/detail/464376
推荐阅读
相关标签
  

闽ICP备14008679号