Mistral Nemo をローカルに導入するには?
Mistral NeMo は、Mistral AI と NVIDIA が 2024 年 7 月に共同で公開した 12B のモデルです。公式は Mistral 7B をそのまま置き換えるモデルと位置づけています。短い会話なら負荷は軽く、4-bit なら 12GB の GPU で足ります。ただしコンテキストは VRAM を大きく消費し、32K 増えるごとに約 5 GB 増え、128K をフルに使うと約 30 GB 必要です。
Mistral Nemo のモデル情報
- 4-bit 最小VRAM10 GB · 8K
- パラメータ数12.3B
- 最大コンテキスト128K
- アーキテクチャ
- Dense
- ライセンス
- 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 のローカル実行に必要なVRAMは?
| 量子化 | 重みサイズ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 |
詳しい計算ルールはデータの作り方ページをご覧ください。
Mistral Nemo は GPU 1 枚で動く?
1 枚で問題なく動き、12GB のカードで 4-bit と 8K コンテキスト(約 9.9 GB)なら十分です。VRAM を食うのはコンテキストで、32K 広げるごとに約 5 GB ずつ増えるため、16GB カードでは 4-bit のままが得策です。8-bit にするとコンテキストの余地がほとんど残りません。質と長さを両立させたいなら、24GB の RTX 3090 の 8-bit で約 69K まで広げられ、速度も快適です。128K をフルに使うには 32GB の RTX 5090 が要ります。なお新しいトークナイザーにより、ソースコードは従来の Mistral モデルより約 3 割少ないトークンで表せます。
| デバイス | 価格 | 速度 t/s | 最大コンテキスト |
|---|---|---|---|
| ¥116,6274 件の出品 | 98.2 | 63K | |
| ¥156,40029 件の出品 | 52.5 | 63K | |
| Radeon RX 7900 XTX24 GB | ¥293,6506 件の出品 | 103.8 | 115K |
| ¥406,9228 件の出品 | 101.6 | 115K | |
| Ryzen AI Max+ 395 128GBマシン全体 · 128 GB | ¥688,8588 件の出品 | 31 | 128K |
| ¥862,1369 件の出品 | 108.2 | 115K | |
| ¥1,399,9909 件の出品 | 172.1 | 128K | |
| ¥1,511,5715 件の出品 | 85.6 | 128K | |
| 価格なし | 172.1 | 128K | |
| Mac Studio M5 Ultra 96GBマシン全体 · 96 GB | 価格なし | 127.7 | 128K |
| 価格なし | 98.2 | 128K | |
| MacBook Pro M5 Max 128GBマシン全体 · 128 GB | 価格なし | 70.1 | 128K |
| 価格なし | 33 | 128K |
| デバイス | 価格 | 速度 t/s | 最大コンテキスト |
|---|---|---|---|
| ¥116,6274 件の出品 | 75.6 | 51K | |
| ¥156,40029 件の出品 | 39.8 | 51K | |
| Radeon RX 7900 XTX24 GB | ¥293,6506 件の出品 | 80.1 | 102K |
| ¥406,9228 件の出品 | 78.3 | 102K | |
| Ryzen AI Max+ 395 128GBマシン全体 · 128 GB | ¥688,8588 件の出品 | 23.3 | 128K |
| ¥862,1369 件の出品 | 83.6 | 102K | |
| ¥1,399,9909 件の出品 | 136.3 | 128K | |
| ¥1,511,5715 件の出品 | 65.6 | 128K | |
| 価格なし | 136.3 | 128K | |
| Mac Studio M5 Ultra 96GBマシン全体 · 96 GB | 価格なし | 99.4 | 128K |
| 価格なし | 75.6 | 128K | |
| MacBook Pro M5 Max 128GBマシン全体 · 128 GB | 価格なし | 53.4 | 128K |
| 価格なし | 24.8 | 128K |
| デバイス | 価格 | 速度 t/s | 最大コンテキスト |
|---|---|---|---|
| ¥116,6274 件の出品 | 47.2 | 18K | |
| ¥156,40029 件の出品 | 24.3 | 18K | |
| Radeon RX 7900 XTX24 GB | ¥293,6506 件の出品 | 50.1 | 69K |
| ¥406,9228 件の出品 | 48.9 | 69K | |
| Ryzen AI Max+ 395 128GBマシン全体 · 128 GB | ¥688,8588 件の出品 | 14.1 | 128K |
| ¥862,1369 件の出品 | 52.4 | 69K | |
| ¥1,399,9909 件の出品 | 88.2 | 120K | |
| ¥1,511,5715 件の出品 | 40.7 | 128K | |
| 価格なし | 88.2 | 128K | |
| Mac Studio M5 Ultra 96GBマシン全体 · 96 GB | 価格なし | 62.9 | 128K |
| 価格なし | 47.2 | 120K | |
| MacBook Pro M5 Max 128GBマシン全体 · 128 GB | 価格なし | 32.9 | 128K |
| 価格なし | 15 | 128K |
よく使われる GPU とマシンだけを、50B 未満と 50B 以上のモデルで分けて、価格の安い順に掲載しています。
Mistral Nemo をローカルで動かす前に知っておきたいこと
temperature は低めに
公式が特に注意しているとおり、NeMo はこれまでの Mistral モデルより低い
temperatureが必要で、推奨は 0.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 を使いましょう。
中国語のトークン効率が高い
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 |