MiniMax-M1-80k / docs /vllm_deployment_guide_cn.md
sriting's picture
update README.md and docs
5856a65

🚀 MiniMax 模型 vLLM 部署指南

📖 简介

我们推荐使用 vLLM 来部署 MiniMax-M1 模型。经过我们的测试,vLLM 在部署这个模型时表现出色,具有以下特点:

  • 🔥 卓越的服务吞吐量性能
  • ⚡ 高效智能的内存管理机制
  • 📦 强大的批量请求处理能力
  • ⚙️ 深度优化的底层性能

MiniMax-M1 模型可在单台配备8个H800或8个H20 GPU的服务器上高效运行。在硬件配置方面,搭载8个H800 GPU的服务器可处理长达200万token的上下文输入,而配备8个H20 GPU的服务器则能够支持高达500万token的超长上下文处理能力。

💾 获取 MiniMax 模型

MiniMax-M1 模型获取

您可以从我们的官方 HuggingFace 仓库下载模型:MiniMax-M1-40kMiniMax-M1-80k

下载命令:

pip install -U huggingface-hub
huggingface-cli download MiniMaxAI/MiniMax-M1-40k
# huggingface-cli download MiniMaxAI/MiniMax-M1-80k

# 如果遇到网络问题,可以设置代理
export HF_ENDPOINT=https://hf-mirror.com

或者使用 git 下载:

git lfs install
git clone https://huggingface.co/MiniMaxAI/MiniMax-M1-40k
git clone https://huggingface.co/MiniMaxAI/MiniMax-M1-80k

⚠️ 重要提示:请确保系统已安装 Git LFS,这对于完整下载模型权重文件是必需的。

🛠️ 部署方案

方案一:使用 Docker 部署(推荐)

为确保部署环境的一致性和稳定性,我们推荐使用 Docker 进行部署。

⚠️ 版本要求

  • 基础要求:vLLM 版本必须 ≥ 0.8.3,以确保对 MiniMax-M1 模型的完整支持
  • 特殊说明:如果使用 vLLM 0.8.3 至 0.9.2 之间的版本,需要修改模型配置文件:
    • 打开 config.json
    • config['architectures'] = ["MiniMaxM1ForCausalLM"] 修改为 config['architectures'] = ["MiniMaxText01ForCausalLM"]
  1. 获取容器镜像:
docker pull vllm/vllm-openai:v0.8.3
  1. 运行容器:
# 设置环境变量
IMAGE=vllm/vllm-openai:v0.8.3
MODEL_DIR=<模型存放路径>
CODE_DIR=<代码路径>
NAME=MiniMaxImage

# Docker运行配置
DOCKER_RUN_CMD="--network=host --privileged --ipc=host --ulimit memlock=-1 --shm-size=2gb --rm --gpus all --ulimit stack=67108864"

# 启动容器
sudo docker run -it \
    -v $MODEL_DIR:$MODEL_DIR \
    -v $CODE_DIR:$CODE_DIR \
    --name $NAME \
    $DOCKER_RUN_CMD \
    $IMAGE /bin/bash

方案二:直接安装 vLLM

如果您的环境满足以下要求:

  • CUDA 12.1
  • PyTorch 2.1

可以直接安装 vLLM

安装命令:

pip install vllm

💡 如果您使用其他环境配置,请参考 vLLM 安装指南

🚀 启动服务

启动 MiniMax-M1 服务

export SAFETENSORS_FAST_GPU=1
export VLLM_USE_V1=0
python3 -m vllm.entrypoints.openai.api_server \
--model <模型存放路径> \
--tensor-parallel-size 8 \
--trust-remote-code \
--quantization experts_int8  \
--max_model_len 4096 \
--dtype bfloat16

API 调用示例

curl http://localhost:8000/v1/chat/completions \
    -H "Content-Type: application/json" \
    -d '{
        "model": "MiniMaxAI/MiniMax-M1",
        "messages": [
            {"role": "system", "content": [{"type": "text", "text": "You are a helpful assistant."}]},
            {"role": "user", "content": [{"type": "text", "text": "Who won the world series in 2020?"}]}
        ]
    }'

❗ 常见问题

模块加载问题

如果遇到以下错误:

import vllm._C  # noqa
ModuleNotFoundError: No module named 'vllm._C'

当前并不支持 MiniMax-M1 模型

我们提供两种解决方案:

解决方案一:复制依赖文件

cd <工作目录>
git clone https://github.com/vllm-project/vllm.git
cd vllm
cp /usr/local/lib/python3.12/dist-packages/vllm/*.so vllm 
cp -r /usr/local/lib/python3.12/dist-packages/vllm/vllm_flash_attn/* vllm/vllm_flash_attn

解决方案二:从源码安装

cd <工作目录>
git clone https://github.com/vllm-project/vllm.git

cd vllm/
pip install -e .

📮 获取支持

如果您在部署 MiniMax-M1 模型过程中遇到任何问题:

  • 请查看我们的官方文档
  • 通过官方渠道联系我们的技术支持团队
  • 在我们的 GitHub 仓库提交 Issue

我们会持续优化模型的部署体验,欢迎您的反馈!