返回顶部
热门问答 更多热门问答
技术文章 更多技术文章

部署 4.8k star 的 Xinference 给知识库接入本地 Rerank 模型

[复制链接]
链载Ai 显示全部楼层 发表于 昨天 11:29 |阅读模式 打印 上一主题 下一主题


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

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

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

简介

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

它支持的模型种类有:

  • 语言模型:比如 qwen2、baichuan、deepseek、gemma 等场景的语言模型

  • Embedding 模型:有 Jina 的 Embedding 模型(结合 Rerank,知识库检索不就起来了嘛?);

  • Rerank 模型:有 Jina 的 Rerank 模型;

  • 图像模型:除了 Stable Diffusion 之外,还有 Flux 模型;

  • 语音模型:有 ChatTTS 以及 whisper 等等;

  • 视频模型:这类模型还没了解过,Xinference 里是 CogVideoX 模型;

  • 自定义模型:需要先注册,然后才可以在这里看到;

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

部署

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

  • 本地运行:也就是使用终端命令行的方式进行部署,需要 Python 环境;

  • 在集群中部署:对于个人用户来说用不上,虽然现在都在上云,但是 Kubernetes 的使用还是多存在于企业中;

  • 使用 Docker 部署:废话不多说,Run 起来!

在使用 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就可以看到这样一个页面:

大概介绍一下界面构成:

  • Launch Model:其实就是目前 Xinference 里内置支持的模型,种类繁多,大家自行选择;

  • Running Models:在 Xinference 中已经下载和运行起来的模型

  • Register Model:对于 Custom Models 来说,需要先注册,才能运行;

  • Cluster Information:在这个页面里可以看到项目运行时的资源消耗情况

实战:安装 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是专业的生成式人工智能教程平台。提供Stable Diffusion、Midjourney AI绘画教程,Suno AI音乐生成指南,以及Runway、Pika等AI视频制作与动画生成实战案例。从提示词编写到参数调整,手把手助您从入门到精通。
  • 官方手机版

  • 微信公众号

  • 商务合作

  • Powered by Discuz! X3.5 | Copyright © 2025-2025. | 链载Ai
  • 桂ICP备2024021734号 | 营业执照 | |广西笔趣文化传媒有限公司|| QQ