为了保证模型推理性能,推荐表1中指定量化版本的Embedding模型,本节以bce-embedding-base_v1模型为例,介绍在鲲鹏CPU平台部署Embedding服务的步骤。
模型名称 |
支持语言 |
参数规模 |
最大输入/tokens |
下载链接 |
|---|---|---|---|---|
bce-embedding-base_v1-f16.gguf |
中文、英文 |
0.28B |
512 |
|
bge-m3-FP16.gguf |
多语言 |
0.57B |
8192 |
|
ritrieve_zh_v1.f16.gguf |
中文 |
0.32B |
8192 |
docker pull swr.cn-north-4.myhuaweicloud.com/kunpeng-ai/llama.cpp:920-kunpeng
docker run \
--cpuset-cpus="0-31" \ # 容器绑核,绑定一个numa上
-v /root/models:/models \ # 挂载模型目录到容器
-p 11133:11133 # 映射容器的服务端口到主机端口
swr.cn-north-4.myhuaweicloud.com/kunpeng-ai/llama.cpp:920-kunpeng \ # 使用已构建好的llama.cpp镜像
--server \ # 启动api服务
-m /models/bce-embedding-base_v1-f16.gguf \ # 指定模型路径
--embedding \
--port 11133 \
--threads 32 \
--host 0.0.0.0 \
--ubatch-size 1024
curl -X POST http://127.0.0.1:11133/v1/embeddings -H "Content-Type: application/json" -d '{
"model": "bce-embedding-base_v1-FP16.gguf",
"input": "Why is the sky blue?"
}'
若能成功返回嵌入向量值,则部署成功。