

docker run -u root -e ASCEND_VISIBLE_DEVICES=0 -itd --name=tei --net=host \ -e HOME=/home/HwHiAiUser \ -e TEI_NPU_DEVICE=0 \[可选] --privileged=true \ -v /root/bge-large-zh-v1.5:/home/HwHiAiUser/model/bge-large-zh-v1.5 \ -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 \[可选,一般不加] swr.cn-south-1.myhuaweicloud.com/ascendhub/mis-tei:6.0.0-300I-Duo-aarch64 BAAI/bge-large-zh-v1.5 ip port
以下参数可根据实际情况进行配置:
在鲲鹏模组RAG一体机部署场景,由于Embedding模型和推理服务部署均部署在NPU上,可能存在Embedding模型占用显存持续增加的问题。处理方式请参见昇腾社区提供的FAQ。
curl ip:port/embed \
-X POST \
-d '{"inputs":"What is Deep Learning?"}' \
-H 'Content-Type: application/json'
