DeL-TaiseiOzaki
commited on
Commit
•
8be9945
1
Parent(s):
0b38a33
Update README.md
Browse files
README.md
CHANGED
@@ -7,4 +7,95 @@ base_model:
|
|
7 |
- llm-jp/llm-jp-3-172b-instruct3
|
8 |
---
|
9 |
|
10 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
- llm-jp/llm-jp-3-172b-instruct3
|
8 |
---
|
9 |
|
10 |
+
# 4-bit 量子化版 llm-jp-3-172b-instruct3
|
11 |
+
|
12 |
+
本リポジトリでは、大学共同利用機関法人情報・システム研究機構 国立情報学研究所(以下「NII」)が提供する「llm-jp-3-172b-instruct3」(以下「本モデル」) を 4-bit 量子化した派生モデル (以下「本量子化モデル」) を公開しています。
|
13 |
+
|
14 |
+
> **Built with llm-jp-3-172b-instruct3**
|
15 |
+
|
16 |
+
---
|
17 |
+
|
18 |
+
## 概要
|
19 |
+
|
20 |
+
- **元モデル**
|
21 |
+
- 名称: [llm-jp-3-172b-instruct3]
|
22 |
+
- 提供元: 大学共同利用機関法人情報・システム研究機構 (ROIS) 国立情報学研究所 (NII)
|
23 |
+
- ライセンス: 「llm-jp-3-172b-instruct3利用許諾契約」 (本リポジトリにも同梱)
|
24 |
+
|
25 |
+
- **本量子化モデルの目的**
|
26 |
+
- 4-bit 量子化を適用し、大規模モデルの推論時における GPU/メモリ使用量の削減を実現するために作成しました。
|
27 |
+
- モデル容量は 4-bit 量子化により削減されますが、内部的には `uint8` + スケール情報を持つ BitsAndBytes (fp4 形式) で格納されており、演算時には float16/float32 を用います。
|
28 |
+
|
29 |
+
---
|
30 |
+
## インストール・利用方法
|
31 |
+
|
32 |
+
1. **環境準備**
|
33 |
+
|
34 |
+
```bash
|
35 |
+
pip install transformers>=4.47 bitsandbytes>=0.39.0
|
36 |
+
```
|
37 |
+
|
38 |
+
2. **モデルのロード例 (Python)**
|
39 |
+
|
40 |
+
```python
|
41 |
+
import torch
|
42 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
43 |
+
from transformers import BitsAndBytesConfig
|
44 |
+
|
45 |
+
# 4-bit 量子化設定
|
46 |
+
quantization_config = BitsAndBytesConfig(
|
47 |
+
load_in_4bit=True,
|
48 |
+
bnb_4bit_compute_dtype=torch.float32, # 演算時には float32
|
49 |
+
bnb_4bit_quant_type="fp4", # FP4 量子化
|
50 |
+
bnb_4bit_quant_storage="uint8"
|
51 |
+
)
|
52 |
+
|
53 |
+
# モデル・トークナイザをロード
|
54 |
+
model_name_or_path = "path/to/4bit-quantized-llm-jp-3-172b-instruct3"
|
55 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code=True)
|
56 |
+
model = AutoModelForCausalLM.from_pretrained(
|
57 |
+
model_name_or_path,
|
58 |
+
quantization_config=quantization_config,
|
59 |
+
device_map="auto",
|
60 |
+
trust_remote_code=True
|
61 |
+
)
|
62 |
+
|
63 |
+
# 推論テスト
|
64 |
+
prompt = "日本語での応答例を示してください。"
|
65 |
+
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
|
66 |
+
output_ids = model.generate(**inputs, max_new_tokens=128)
|
67 |
+
print(tokenizer.decode(output_ids[0], skip_special_tokens=True))
|
68 |
+
```
|
69 |
+
|
70 |
+
3. **GPU メモリ削減効果の確認 (オプション)**
|
71 |
+
- 量子化モデルをロードした際の `torch.cuda.memory_allocated()` を確認すると、FP16/FP32 モデルよりもメモリ使用量が削減されているはずです。
|
72 |
+
|
73 |
+
---
|
74 |
+
|
75 |
+
## ライセンス
|
76 |
+
|
77 |
+
本量子化モデルは、**元モデル「llm-jp-3-172b-instruct3」** を利用し作成した派生モデルです。
|
78 |
+
以下の条件に従い利用が可能です。
|
79 |
+
|
80 |
+
1. 本量子化モデルに含まれる元モデル部分に関しては、**「[llm-jp-3-172b-instruct3利用許諾契約](./LICENSE)」** (以下「本契約」) に準拠します。
|
81 |
+
2. 本量子化モデルを再配布または利用可能にする場合は、**本契約書のコピーを添付** し、かつドキュメント等に下記の表示を行ってください:
|
82 |
+
> 「Built with llm-jp-3-172b-instruct3」
|
83 |
+
3. 本量子化モデル内に含まれる元モデルとその派生モデル部分に対しては、本契約に記載のとおり **著作権表示** を明示してください:
|
84 |
+
> 「llm-jp-3-172b-instruct3 は llm-jp-3-172b-instruct3利用許諾契約に基づきライセンスされています。
|
85 |
+
> Copyright (c) 2024 Research Organization of Information and Systems.
|
86 |
+
> すべての権利を留保します。」
|
87 |
+
|
88 |
+
---
|
89 |
+
|
90 |
+
## 注意事項・免責
|
91 |
+
|
92 |
+
- **本量子化モデルは「現状有姿」で提供されます**。元モデルと同様に、正確性、完全性、商品性、特定目的への適合性など、いかなる保証も行いません。
|
93 |
+
- 本モデルおよびその出力の利用により生じる結果や損害について、開発者・提供者は責任を負いかねます。
|
94 |
+
- 詳細は、同梱の「llm-jp-3-172b-instruct3利用許諾契約」をご参照ください。
|
95 |
+
---
|
96 |
+
|
97 |
+
## 謝辞
|
98 |
+
|
99 |
+
- 本モデルの開発にあたり、大学共同利用機関法人情報・システム研究機構 国立情報学研究所 (ROIS/NII) が提供する大規模言語モデル「llm-jp-3-172b-instruct3」を使用しました。
|
100 |
+
|
101 |
+
---
|