昇腾300I npu部署bge-m3 embedding模型
·
本次任务将使用mis-tei镜像部署bge-m3模型
一、下载mis-tei镜像,该镜像下载需提前申请。

二、使用docker启动镜像
docker run -u root -e ASCEND_VISIBLE_DEVICES=4 -itd --name=bge-m3 --net=host \
-e HOME=/home/HwHiAiUser \
--privileged=true \
-v /home/BAAI/:/home/HwHiAiUser/model \ #把整个模型目录放在/home/BAAI目录里
-v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
-v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
--entrypoint /home/HwHiAiUser/start.sh \
mis-tei:6.0.0-300I-Duo-aarch64 \
BAAI/bge-m3 127.0.0.1 8086 #可以改为本机ip,外部才可请求
https://www.hikunpeng.com/doc_center/source/zh/kunpengrag/bestpractice/kunpengrag_21_0014.html
三、请求测试
curl http://127.0.0.1:8086/v1/embeddings \
-H "Content-Type: application/json" \
-d '{
"input": "test text",
"model": "bge-m3"
}'
如有疏漏,敬请指正
昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链
更多推荐

所有评论(0)