|
# 🚀 MiniMax 模型 vLLM 部署指南 |
|
|
|
## 📖 简介 |
|
|
|
我们推荐使用 [vLLM](https://docs.vllm.ai/en/latest/) 来部署 [MiniMax-M1](https://huggingface.co/MiniMaxAI/MiniMax-M1-40k) 模型。经过我们的测试,vLLM 在部署这个模型时表现出色,具有以下特点: |
|
|
|
- 🔥 卓越的服务吞吐量性能 |
|
- ⚡ 高效智能的内存管理机制 |
|
- 📦 强大的批量请求处理能力 |
|
- ⚙️ 深度优化的底层性能 |
|
|
|
MiniMax-M1 模型可在单台配备8个H800或8个H20 GPU的服务器上高效运行。在硬件配置方面,搭载8个H800 GPU的服务器可处理长达200万token的上下文输入,而配备8个H20 GPU的服务器则能够支持高达500万token的超长上下文处理能力。 |
|
|
|
## 💾 获取 MiniMax 模型 |
|
|
|
### MiniMax-M1 模型获取 |
|
|
|
您可以从我们的官方 HuggingFace 仓库下载模型:[MiniMax-M1-40k](https://huggingface.co/MiniMaxAI/MiniMax-M1-40k)、[MiniMax-M1-80k](https://huggingface.co/MiniMaxAI/MiniMax-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 下载: |
|
|
|
```bash |
|
git lfs install |
|
git clone https://huggingface.co/MiniMaxAI/MiniMax-M1-40k |
|
git clone https://huggingface.co/MiniMaxAI/MiniMax-M1-80k |
|
``` |
|
|
|
⚠️ **重要提示**:请确保系统已安装 [Git LFS](https://git-lfs.github.com/),这对于完整下载模型权重文件是必需的。 |
|
|
|
## 🛠️ 部署方案 |
|
|
|
### 方案一:使用 Docker 部署(推荐) |
|
|
|
为确保部署环境的一致性和稳定性,我们推荐使用 Docker 进行部署。 |
|
|
|
⚠️ **版本要求**: |
|
- 基础要求:vLLM 版本必须 ≥ 0.8.3,以确保对 MiniMax-M1 模型的完整支持 |
|
- 特殊说明:如果使用 vLLM 0.8.3 至 0.9.2 之间的版本,需要修改模型配置文件: |
|
- 打开 `config.json` |
|
- 将 `config['architectures'] = ["MiniMaxM1ForCausalLM"]` 修改为 `config['architectures'] = ["MiniMaxText01ForCausalLM"]` |
|
|
|
1. 获取容器镜像: |
|
```bash |
|
docker pull vllm/vllm-openai:v0.8.3 |
|
``` |
|
|
|
2. 运行容器: |
|
```bash |
|
# 设置环境变量 |
|
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 |
|
|
|
安装命令: |
|
```bash |
|
pip install vllm |
|
``` |
|
|
|
💡 如果您使用其他环境配置,请参考 [vLLM 安装指南](https://docs.vllm.ai/en/latest/getting_started/installation.html) |
|
|
|
## 🚀 启动服务 |
|
|
|
### 启动 MiniMax-M1 服务 |
|
|
|
```bash |
|
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 调用示例 |
|
|
|
```bash |
|
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 模型 |
|
``` |
|
|
|
我们提供两种解决方案: |
|
|
|
#### 解决方案一:复制依赖文件 |
|
```bash |
|
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 |
|
``` |
|
|
|
#### 解决方案二:从源码安装 |
|
```bash |
|
cd <工作目录> |
|
git clone https://github.com/vllm-project/vllm.git |
|
|
|
cd vllm/ |
|
pip install -e . |
|
``` |
|
|
|
## 📮 获取支持 |
|
|
|
如果您在部署 MiniMax-M1 模型过程中遇到任何问题: |
|
- 请查看我们的官方文档 |
|
- 通过官方渠道联系我们的技术支持团队 |
|
- 在我们的 GitHub 仓库提交 [Issue](https://github.com/MiniMax-AI/MiniMax-M1/issues) |
|
|
|
我们会持续优化模型的部署体验,欢迎您的反馈! |
|
|