1. 下载词向量模型
参考我的另一篇博客:langchain 加载本地词向量模型
2. 部署词向量模型
就三行命令
model=/data/BAAI/bge-m3
volume=/home/project/text_embedding/models
docker run -p 8082:80 -v $volume:/data --pull always ghcr.io/huggingface/text-embeddings-inference:cpu-1.5 --model-id $model
会先去下载 docker 镜像,然后运行容器。我这里下载的是 cpu 版本的 docker 镜像,其他硬件环境的 docker 镜像参考:https://huggingface.co/docs/text-embeddings-inference/supported_models
特别要强调的是:model 的路径可不是随便写的。在我的这个例子中,启动 docker 时,映射的路径是**
$volume:/data
** ,因此 model 的路径必须以**/data** 开头,不然的话是找不到模型的。我的模型存放路径:
/home/project/text_embedding/models/BAAI/bge-m3
,大家可以参考我的路径来调整 model 和 volume 变量的值。
3. 调用词向量模型
使用 REST API 调用,可调用的 API 参考:Text Embeddings Inference API
curl 127.0.0.1:8082/embed \
-X POST \
-d '{"inputs":"What is Deep Learning?"}' \
-H 'Content-Type: application/json'