链载Ai

标题: 部署 4.8k star 的 Xinference 给知识库接入本地 Rerank 模型 [打印本页]

作者: 链载Ai    时间: 昨天 11:29
标题: 部署 4.8k star 的 Xinference 给知识库接入本地 Rerank 模型


首先说说为什么会想要部署 Xinference

起因是因为前几天白嫖的 Jina API token 用完了,而付费的话又感觉自己的那些知识库没有必要使用商用的(好吧好吧,其实还是舍不得花钱?),开源免费的对我来说也足够了,所以想要找找可以本地部署的。

刚开始想使用 Ollama 的,不过查了下它好像不支持 Rerank 模型,因此又找上了 Xinference 这款产品。

简介

Xorbits Inference (Xinference) 是一个开源平台,用于简化各种 AI 模型的运行和集成。借助 Xinference,我们可以使用任何开源 LLM、嵌入模型和多模态模型在云端或本地环境中运行推理,并创建强大的 AI 应用。

它支持的模型种类有:

相较于 Ollama 来说,Xinference 在部署之后会为我们提供一个可视化界面,我们可以通过图形化界面安装部署大模型,这个下来会详细进行讲解。

部署

官方介绍了三种部署方式,分别是:

在使用 Docker 部署 Xinference 时,如果你的机器有 GPU,可以执行(把your_version改为你想运行的镜像版本,比如 latest):

docker run -e XINFERENCE_MODEL_SRC=modelscope -p 9998:9997 --gpus all xprobe/xinference:<your_version> xinference-local -H 0.0.0.0 --log-level debug

如果是只有 CPU 的机器,则执行:

docker run -e XINFERENCE_MODEL_SRC=modelscope -p 9998:9997 xprobe/xinference:<your_version>-cpu xinference-local -H 0.0.0.0 --log-level debug

在启动容器之后,我们访问localhost:9998就可以看到这样一个页面:

大概介绍一下界面构成:

实战:安装 Rerank 模型并集成到 Dify 中

在 RERANK MODELS 中选择一个模型进行部署启动,以bce-reranker-base_v1为例:

点击左下角的那个小火箭开启下载~

下载好之后,可以到 Running Models 中的 RERANK MODELS 里看到对应的模型:

我们到 Dify 中集成一下 Xinference(在模型供应商的地方):

因为我们下载的是 Rerank 模型,所以这里「模型类型」为 Rerank 模型;

模型名称和模型 UID 分别填入下图中红框框住的部分(ID 对应 UID,名称对应 Name):

服务器 URL 这里填http://host.docker.internal:9998,这样写的原因在之前的文章中《Dify 教程二:使用本地大模型 Ollama》也说过。如果不是 Docker 部署而是本地运行或者集群部署则可以直接写「服务器 IP:端口号」。

保存!开测~

在 Dify 中新建一个知识库,并上传数据集(本次上传了《断舍离》这本书进行测试),知识库的检索设置如下:

最终得到 261 个分段,召回测试如下:

我们将这个知识库加到应用中试试看:

效果还不错,感兴趣的小伙伴可以试试看。






欢迎光临 链载Ai (https://www.lianzai.com/) Powered by Discuz! X3.5