Skip to content

Latest commit

 

History

History
267 lines (215 loc) · 12.5 KB

README_zh_CN.md

File metadata and controls

267 lines (215 loc) · 12.5 KB
xorbits

Xorbits Inference:模型推理, 轻而易举 🤖

PyPI Latest Release License Build Status Slack Twitter

English | 中文介绍 | 日本語


Xorbits Inference(Xinference)是一个性能强大且功能全面的分布式推理框架。可用于大语言模型(LLM),语音识别模型,多模态模型等各种模型的推理。通过 Xorbits Inference,你可以轻松地一键部署你自己的模型或内置的前沿开源模型。无论你是研究者,开发者,或是数据科学家,都可以通过 Xorbits Inference 与最前沿的 AI 模型,发掘更多可能。

🔥 近期热点

框架增强

  • 支持 rerank 模型: #672
  • 支持指定 grammar 输出: #525
  • 投机采样: #509
  • 引入 vLLM: #445

新模型

集成

  • Dify: 一个涵盖了大型语言模型开发、部署、维护和优化的 LLMOps 平台。
  • Chatbox: 一个支持前沿大语言模型的桌面客户端,支持 Windows,Mac,以及 Linux。

主要功能

🌟 模型推理,轻而易举:大语言模型,语音识别模型,多模态模型的部署流程被大大简化。一个命令即可完成模型的部署工作。

⚡️ 前沿模型,应有尽有:框架内置众多中英文的前沿大语言模型,包括 baichuan,chatglm2 等,一键即可体验!内置模型列表还在快速更新中!

🖥 异构硬件,快如闪电:通过 ggml,同时使用你的 GPU 与 CPU 进行推理,降低延迟,提高吞吐!

⚙️ 接口调用,灵活多样:提供多种使用模型的接口,包括 RPC,RESTful API,命令行,web UI 等等。方便模型的管理与监控。

🌐 集群计算,分布协同: 支持分布式部署,通过内置的资源调度器,让不同大小的模型按需调度到不同机器,充分使用集群资源。

🔌 开放生态,无缝对接: 与流行的三方库无缝对接,包括 LangChainLlamaIndexDify,以及 Chatbox

为什么选择 Xinference

功能特点 Xinference FastChat OpenLLM RayLLM
兼容 OpenAI 的 Restful API
vLLM 集成
更多推理引擎(GGML、TensorRT)
更多平台支持(CPU、Metal)
分布式集群部署
多模态模型(文生图)
文本嵌入模型

快速入门

安装

Xinference 可以通过 pip 从 PyPI 安装。我们非常推荐在安装前创建一个新的虚拟环境以避免依赖冲突。

使用 Xinference 前,您需要安装与模型类型相对应的后端。如果想要推理所有支持的模型,可以安装所有后端:

pip install "xinference[all]"

注意:推理 GGML 格式的模型前,我们强烈建议手动安装 GGML 依赖,以在不同硬件上达到加速效果,请参考 GGML 后端

Transformers 后端

Transformers 后端支持绝大多数前沿模型,它是 PyTorch 格式模型的默认后端:

pip install "xinference[transformers]"

vLLM 后端

vLLM 后端能够提供高效的推理能力。当满足以下条件时,Xinference 会选择 vLLM 作为后端以达到更好的吞吐量:

  • 模型格式为 PyTorch
  • 模型在下面的支持列表中
  • 量化选择 none(AWQ 量化将会在近期支持)
  • Linux 系统,并有 CUDA 设备

目前, 支持的模型包括:

  • llama-2, llama-2-chat
  • baichuan, baichuan-chat
  • internlm, internlm-20b, internlm-chat, internlm-chat-20b
  • vicuna-v1.3, vicuna-v1.5
pip install "xinference[vllm]"

GGML 后端

我们强烈建议手动安装 GGML 依赖,以在不同硬件上达到加速效果。

初始安装:

pip install xinference
pip install ctransformers

根据硬件,选择性安装:

  • 苹果芯片(M1,M2):
    CMAKE_ARGS="-DLLAMA_METAL=on" pip install llama-cpp-python
  • 英伟达 GPU:
    CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip install llama-cpp-python
  • AMD GPU:
    CMAKE_ARGS="-DLLAMA_HIPBLAS=on" pip install llama-cpp-python

部署

你可以一键进行本地部署,或按照下面的步骤将 Xinference 部署在计算集群。

本地部署

运行下面的命令在本地部署 Xinference:

$ xinference-local

分布式部署

分布式场景下,你需要在一台服务器上部署一个 Xinference supervisor,并在其余服务器上分别部署一个 Xinference worker。 具体步骤如下:

启动 supervisor: 执行:

$ xinference-supervisor -H "${supervisor_host}"

替换 ${supervisor_host} 为 supervisor 所在服务器的实际主机名或 IP 地址。

启动 workers: 在其余服务器上,执行:

$ xinference-worker -e "http://${supervisor_host}:9997"

Xinference 启动后,将会打印服务的 endpoint。这个 endpoint 用于通过命令行工具或编程接口进行模型的管理。

  • 本地部署下, endpoint 默认为 http://localhost:9997.
  • 集群部署下, endpoint 默认为 http://${supervisor_host}:9997。其中 ${supervisor_host} 为supervisor 所在服务器的主机名或 IP 地址。

你还可以通过 web UI 启动并管理模型,使用任意内置模型生成文本或对话。

web UI

Xinference 命令行

Xinference 提供了命令行工具用于模型管理。支持的命令包括:

  • 启动一个模型 (将会返回一个模型 UID):xinference launch
  • 查看所有运行中的模型:xinference list
  • 查看所有支持的模型:xinference registrations
  • 结束模型:xinference terminate --model-uid ${model_uid}

Xinference 编程接口

Xinference 同样提供了编程接口:

from xinference.client import Client

client = Client("http://localhost:9997")
model_uid = client.launch_model(model_name="chatglm2")
model = client.get_model(model_uid)

chat_history = []
prompt = "What is the largest animal?"
model.chat(
    prompt,
    chat_history,
    generate_config={"max_tokens": 1024}
)

返回值:

{
  "id": "chatcmpl-8d76b65a-bad0-42ef-912d-4a0533d90d61",
  "model": "56f69622-1e73-11ee-a3bd-9af9f16816c6",
  "object": "chat.completion",
  "created": 1688919187,
  "choices": [
    {
      "index": 0,
      "message": {
        "role": "assistant",
        "content": "The largest animal that has been scientifically measured is the blue whale, which has a maximum length of around 23 meters (75 feet) for adult animals and can weigh up to 150,000 pounds (68,000 kg). However, it is important to note that this is just an estimate and that the largest animal known to science may be larger still. Some scientists believe that the largest animals may not have a clear \"size\" in the same way that humans do, as their size can vary depending on the environment and the stage of their life."
      },
      "finish_reason": "None"
    }
  ],
  "usage": {
    "prompt_tokens": -1,
    "completion_tokens": -1,
    "total_tokens": -1
  }
}

请参考 更多案例

内置模型

运行以下命令查看内置模型列表:

$ xinference registrations
Type Name Language Ability
LLM baichuan ['en', 'zh'] ['embed', 'generate']
LLM baichuan-2 ['en', 'zh'] ['embed', 'generate']
LLM baichuan-chat ['en', 'zh'] ['embed', 'chat']
LLM baichuan-2-chat ['en', 'zh'] ['embed', 'chat']
LLM chatglm ['en', 'zh'] ['embed', 'chat']
LLM chatglm2 ['en', 'zh'] ['embed', 'chat']
LLM chatglm2-32k ['en', 'zh'] ['embed', 'chat']
LLM code-llama ['en'] ['generate']
LLM code-llama-instruct ['en'] ['chat']
LLM code-llama-python ['en'] ['generate']
LLM falcon ['en'] ['embed', 'generate']
LLM falcon-instruct ['en'] ['embed', 'chat']
LLM glaive-coder ['en'] ['chat']
LLM gpt-2 ['en'] ['generate']
LLM internlm-7b ['en', 'zh'] ['embed', 'generate']
LLM internlm-chat-7b ['en', 'zh'] ['embed', 'chat']
LLM internlm-chat-20b ['en', 'zh'] ['embed', 'chat']
LLM llama-2 ['en'] ['embed', 'generate']
LLM llama-2-chat ['en'] ['embed', 'chat']
LLM opt ['en'] ['embed', 'generate']
LLM orca ['en'] ['embed', 'chat']
LLM qwen-chat ['en', 'zh'] ['embed', 'chat']
LLM starchat-beta ['en'] ['embed', 'chat']
LLM starcoder ['en'] ['generate']
LLM starcoderplus ['en'] ['embed', 'generate']
LLM vicuna-v1.3 ['en'] ['embed', 'chat']
LLM vicuna-v1.5 ['en'] ['embed', 'chat']
LLM vicuna-v1.5-16k ['en'] ['embed', 'chat']
LLM wizardlm-v1.0 ['en'] ['embed', 'chat']
LLM wizardmath-v1.0 ['en'] ['embed', 'chat']
LLM OpenBuddy ['en', 'zh'] ['embed', 'chat']

更多信息请参考 内置模型

注意:

  • Xinference 会自动为你下载模型,默认的模型存放路径为 ${USER}/.xinference/cache
  • 如果您在Hugging Face下载模型时遇到问题,请运行 export XINFERENCE_MODEL_SRC=modelscope,默认优先从 modelscope 下载。目前 modelscope 支持的模型有:
    • llama-2
    • llama-2-chat
    • baichuan-2
    • baichuan-2-chat
    • chatglm2
    • chatglm2-32k
    • internlm-chat-20b
    • ... 更多的支持的模型在文档中列出了。

自定义模型

请参考 自定义模型