Mistral Nemo 如何本地部署?
Mistral NeMo 是 Mistral AI 和 NVIDIA 在 2024 年 7 月合作发布的 12B 模型,官方定位是 Mistral 7B 的直接替代。短对话很轻,4-bit 在 12GB 显卡上就够;但它的上下文很吃显存,每多 32K 约多 5 GB,开满 128K 要约 30 GB。
Mistral Nemo 模型信息
- 4-bit 最低显存10 GB · 8K
- 参数量12.3B
- 上下文上限128K
- 架构
- 稠密
- 许可证
- Apache-2.0
- 发布日期
- 2024年7月17日
- 厂商
- Mistral AI
- Hugging Face 仓库
- mistralai/Mistral-Nemo-Instruct-2407
- MLX 仓库(Apple)
- mlx-community/Mistral-Nemo-Instruct-2407-4bit
- Ollama 标签
- mistral-nemo:12b
- 小版本
- 2407
- 层数
- 40
- 隐藏维度
- 5,120
- KV 头
- 8
- 头维度
- 128
Mistral Nemo 本地部署需要多少显存?
| 量化 | 权重体积GB | 8K 上下文GB | 32K 上下文GB | 128K 上下文(最大)GB |
|---|---|---|---|---|
| 1-bit | 3.1估 | 5.5 | 9.5 | 25.6 |
| 2-bit | 5.4GGUF | 7.8 | 11.8 | 27.9 |
| 3-bit | 7.1GGUF | 9.5 | 13.5 | 29.6 |
| 4-bit | 7.5GGUF | 9.9 | 13.9 | 30 |
| 5-bit | 8.7GGUF | 11.1 | 15.1 | 31.2 |
| 8-bit | 13GGUF | 15.4 | 19.4 | 35.5 |
| BF16/FP16 | 24.5GGUF | 26.9 | 30.9 | 47 |
详细计算规则可在数据方法页查看。
| 设备 | 价格 | 速度 t/s | 最大上下文 |
|---|---|---|---|
| ¥1,2505 条挂单 | 98.2 | 63K | |
| ¥4,2004 条挂单 | 98.2 | 128K | |
| ¥5,17514 条挂单 | 52.5 | 63K | |
| Radeon RX 7900 XTX24 GB | ¥5,69412 条挂单 | 103.8 | 115K |
| ¥8,50013 条挂单 | 101.6 | 115K | |
| Ryzen AI Max+ 395 128GB整机 · 128 GB | ¥19,00016 条挂单 | 31 | 128K |
| ¥25,0006 条挂单 | 108.2 | 115K | |
| ¥32,0007 条挂单 | 85.6 | 128K | |
| ¥43,00041 条挂单 | 33 | 128K | |
| MacBook Pro M5 Max 128GB整机 · 128 GB | ¥45,69410 条挂单 | 70.1 | 128K |
| Mac Studio M5 Ultra 96GB整机 · 96 GB | ¥46,00015 条挂单 | 127.7 | 128K |
| ¥51,99914 条挂单 | 172.1 | 128K | |
| ¥125,0006 条挂单 | 172.1 | 128K |
| 设备 | 价格 | 速度 t/s | 最大上下文 |
|---|---|---|---|
| ¥1,2505 条挂单 | 75.6 | 51K | |
| ¥4,2004 条挂单 | 75.6 | 128K | |
| ¥5,17514 条挂单 | 39.8 | 51K | |
| Radeon RX 7900 XTX24 GB | ¥5,69412 条挂单 | 80.1 | 102K |
| ¥8,50013 条挂单 | 78.3 | 102K | |
| Ryzen AI Max+ 395 128GB整机 · 128 GB | ¥19,00016 条挂单 | 23.3 | 128K |
| ¥25,0006 条挂单 | 83.6 | 102K | |
| ¥32,0007 条挂单 | 65.6 | 128K | |
| ¥43,00041 条挂单 | 24.8 | 128K | |
| MacBook Pro M5 Max 128GB整机 · 128 GB | ¥45,69410 条挂单 | 53.4 | 128K |
| Mac Studio M5 Ultra 96GB整机 · 96 GB | ¥46,00015 条挂单 | 99.4 | 128K |
| ¥51,99914 条挂单 | 136.3 | 128K | |
| ¥125,0006 条挂单 | 136.3 | 128K |
| 设备 | 价格 | 速度 t/s | 最大上下文 |
|---|---|---|---|
| ¥1,2505 条挂单 | 47.2 | 18K | |
| ¥4,2004 条挂单 | 47.2 | 120K | |
| ¥5,17514 条挂单 | 24.3 | 18K | |
| Radeon RX 7900 XTX24 GB | ¥5,69412 条挂单 | 50.1 | 69K |
| ¥8,50013 条挂单 | 48.9 | 69K | |
| Ryzen AI Max+ 395 128GB整机 · 128 GB | ¥19,00016 条挂单 | 14.1 | 128K |
| ¥25,0006 条挂单 | 52.4 | 69K | |
| ¥32,0007 条挂单 | 40.7 | 128K | |
| ¥43,00041 条挂单 | 15 | 128K | |
| MacBook Pro M5 Max 128GB整机 · 128 GB | ¥45,69410 条挂单 | 32.9 | 128K |
| Mac Studio M5 Ultra 96GB整机 · 96 GB | ¥46,00015 条挂单 | 62.9 | 128K |
| ¥51,99914 条挂单 | 88.2 | 120K | |
| ¥125,0006 条挂单 | 88.2 | 128K |
只列常见的显卡和整机,50B 以下和以上的模型各一组,按价格从低到高。
本地部署 Mistral Nemo 要注意什么?
温度要调低
官方特别提醒,NeMo 比之前的 Mistral 模型需要更低的温度,建议
temperature0.3。很多聊天前端默认 0.7 以上,记得手动改。量化选哪一档
官方说训练时就按量化做过适配,用 FP8 推理不掉精度。GGUF 的 4-bit 约 7.5 GB、5-bit 约 8.7 GB、8-bit 约 13.0 GB:12GB 卡用 4-bit 或 5-bit,16GB 以上再上 8-bit。
中文更省 token
NeMo 换了新的 Tekken 分词器,官方称中文和代码的压缩效率比之前高约 30%,也就是同样 128K 的上下文能装进更多中文。
用什么框架
llama.cpp、vLLM 直接支持,Ollama 标签是 mistral-nemo:12b。它用的是标准结构,原来跑 Mistral 7B 的工具链基本不用改。
启动命令
Ollama
- Windows
- macOS
- Linux
ollama run mistral-nemo:12bllama.cpp
- NVIDIA
- AMD
- Mac
- CPU
llama-server -hf bartowski/Mistral-Nemo-Instruct-2407-GGUF:Q4_K_M \ -c 8192
vLLM
- Linux
- NVIDIA
- AMD
vllm serve mistralai/Mistral-Nemo-Instruct-2407MLX
- Apple 芯片
Mac 上先 pip install mlx-lm,这条命令会从 Hugging Face 下载 MLX 权重,并启动兼容 OpenAI 接口的服务。
mlx_lm.server --model mlx-community/Mistral-Nemo-Instruct-2407-4bit量化文件名以模型库当周抓取为准,口径见数据方法。 数据方法
Mistral 家族的其他模型
| 模型 | 总参数 | 激活参数 | 4-bit 模型大小 |
|---|---|---|---|
| 7.3B | 7.3B | 4.4 GB | |
| 8.4B | 8.4B | 5.2 GB | |
| 13.5B | 13.5B | 8.2 GB | |
| 24B | 24B | 14.3 GB |