赞
踩
流程记录
阿里云GPU主机,GPU是A10、24G显存
1、CodeGeex2,参数文件,可以在国内镜像网站下载
2、api-for-open-llm,https://github.com/xusenlinzy/api-for-open-llm
建立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 transformers4.33.2
8、测试,http://ip:8000/v1/models (我做的内网穿透,ip是代理地址)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。