kanhatakeyama
commited on
Commit
•
adc8eaa
1
Parent(s):
12e11bb
Update README.md
Browse files
README.md
CHANGED
@@ -12,16 +12,14 @@ tags:
|
|
12 |
|
13 |
## モデルについて
|
14 |
|
15 |
-
Tanuki-8B
|
16 |
-
Tanuki-
|
17 |
-
|
|
|
|
|
18 |
|
19 |
## 量子化モデル
|
20 |
-
|
21 |
-
- [GPTQ 4bit量子化](https://huggingface.co/team-hatakeyama-phase2/Tanuki-8B-dpo-v1.0-GPTQ-4bit)
|
22 |
-
- [GPTQ 8bit量子化](https://huggingface.co/team-hatakeyama-phase2/Tanuki-8B-dpo-v1.0-GPTQ-8bit)
|
23 |
-
- [GGUF量子化](https://huggingface.co/team-hatakeyama-phase2/Tanuki-8B-dpo-v1.0-GGUF)*
|
24 |
-
|
25 |
*GGUF版は性能低下が発生している可能性があり、非推奨
|
26 |
|
27 |
## 使い方
|
@@ -35,7 +33,7 @@ streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
|
|
35 |
|
36 |
messages = [
|
37 |
{"role": "system", "content": "以下は、タスクを説明する指示です。要求を適切に満たす応答を書きなさい。"},
|
38 |
-
{"role": "user", "content": "
|
39 |
]
|
40 |
|
41 |
input_ids = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to(model.device)
|
@@ -46,23 +44,45 @@ output_ids = model.generate(input_ids,
|
|
46 |
```
|
47 |
|
48 |
## プロンプト形式
|
49 |
-
Tanuki-8B-dpo-v1.0は日本語版Alpaca
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
-
|
54 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
55 |
|
56 |
-
|
57 |
-
|
58 |
-
```
|
59 |
|
60 |
-
|
|
|
|
|
|
|
61 |
|
62 |
-
|
63 |
|
64 |
-
|
65 |
-
GPT-4による評価 (Nejumi Leaderboard Neo、gpt-4-0613、平均スコア算出においてスコア-1は除外)
|
66 |
|
67 |
| | Tanuki-8B-dpo-v1.0 | Tanuki-8x8B-dpo-v1.0 |
|
68 |
| ---- | ---- | ---- |
|
@@ -76,10 +96,5 @@ GPT-4による評価 (Nejumi Leaderboard Neo、gpt-4-0613、平均スコア算
|
|
76 |
| stem | 9.35 | 9.40 |
|
77 |
| writing | 9.05 | 8.85 |
|
78 |
|
79 |
-
|
80 |
-
|
81 |
-
|
82 |
-
**人手評価**
|
83 |
-
Chatbot Arena的なシステムを使った人手によるブラインドテストの結果
|
84 |
-
![image/png](https://cdn-uploads.huggingface.co/production/uploads/6348501e50fe0799927c3667/RzPOQfETYD9_AFEjVkwCX.png)
|
85 |
-
|
|
|
12 |
|
13 |
## モデルについて
|
14 |
|
15 |
+
Tanuki-8Bは、フルスクラッチで約1.3Tトークン事前学習を行った約8Bパラメータの大規模言語モデルです。
|
16 |
+
Tanuki-8x8B-dpo-v1.0は、SFTおよびDPOにより対話用に調整されています。
|
17 |
+
より詳細な情報については[ブログ記事](https://zenn.dev/matsuolab/articles/377f7ae8b1169e)を参照してください。
|
18 |
+
|
19 |
+
[GENIAC 松尾研 LLM開発プロジェクト](https://weblab.t.u-tokyo.ac.jp/geniac_llm/)のもと、一般公募により集まった有志の参加者(企業所属、学生、研究者等)が開発を実施しました。
|
20 |
|
21 |
## 量子化モデル
|
22 |
+
[AWQ 4bit量子化](https://huggingface.co/team-hatakeyama-phase2/Tanuki-8B-dpo-v1.0-AWQ) [GPTQ 4bit量子化](https://huggingface.co/team-hatakeyama-phase2/Tanuki-8B-dpo-v1.0-GPTQ-4bit) [GPTQ 8bit量子化](https://huggingface.co/team-hatakeyama-phase2/Tanuki-8B-dpo-v1.0-GPTQ-8bit) [GGUF量子化](https://huggingface.co/team-hatakeyama-phase2/Tanuki-8B-dpo-v1.0-GGUF)*
|
|
|
|
|
|
|
|
|
23 |
*GGUF版は性能低下が発生している可能性があり、非推奨
|
24 |
|
25 |
## 使い方
|
|
|
33 |
|
34 |
messages = [
|
35 |
{"role": "system", "content": "以下は、タスクを説明する指示です。要求を適切に満たす応答を書きなさい。"},
|
36 |
+
{"role": "user", "content": "たぬきに純粋理性批判は理解できますか?"}
|
37 |
]
|
38 |
|
39 |
input_ids = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to(model.device)
|
|
|
44 |
```
|
45 |
|
46 |
## プロンプト形式
|
47 |
+
Tanuki-8B-dpo-v1.0は日本語版Alpacaのプロンプト形式を利用します。
|
48 |
+
- シングルターン
|
49 |
+
```
|
50 |
+
<s>以下は、タスクを説明する指示です。要求を適切に満たす応答を書きなさい。
|
51 |
+
|
52 |
+
### 指示:
|
53 |
+
たぬきに純粋理性批判は理解できますか?
|
54 |
+
|
55 |
+
### 応答:
|
56 |
+
|
57 |
+
```
|
58 |
+
- マルチターン
|
59 |
+
```
|
60 |
+
<s>以下は、タスクを説明する指示です。要求を適切に満たす応答を書きなさい。
|
61 |
+
|
62 |
+
### 指示:
|
63 |
+
{1ターン目の入力}
|
64 |
+
|
65 |
+
### 応答:
|
66 |
+
{1ターン目の応答}</s>
|
67 |
+
|
68 |
+
### 指示:
|
69 |
+
{2ターン目の入力}
|
70 |
+
|
71 |
+
### 応答:
|
72 |
+
```
|
73 |
+
|
74 |
+
なお、本モデルはデフォルトのシステムプロンプトである「以下は、タスクを説明する指示です。要求を適切に満たす応答を書きなさい。」以外を学習していないため、このシステムプロンプトの使用を推奨します。タスクの詳細はユーザープロンプトに記述してください。
|
75 |
|
76 |
+
## ベンチマーク
|
|
|
|
|
77 |
|
78 |
+
**人手評価**
|
79 |
+
Chatbot Arenaを模したシステムを作成し、人手によるブラインドテストを実施しました。(詳細は[こちら](https://zenn.dev/matsuolab/articles/95fa297ef12a14))
|
80 |
+
評価結果(約2000件)を後日全て公開します。
|
81 |
+
![image/png](https://cdn-uploads.huggingface.co/production/uploads/6348501e50fe0799927c3667/RzPOQfETYD9_AFEjVkwCX.png)
|
82 |
|
83 |
+
**Japanese MT-Bench**
|
84 |
|
85 |
+
GPT-4による評価 (gpt-4-0613、平均スコア算出においてスコア-1は除外)
|
|
|
86 |
|
87 |
| | Tanuki-8B-dpo-v1.0 | Tanuki-8x8B-dpo-v1.0 |
|
88 |
| ---- | ---- | ---- |
|
|
|
96 |
| stem | 9.35 | 9.40 |
|
97 |
| writing | 9.05 | 8.85 |
|
98 |
|
99 |
+
## 開発メンバー
|
100 |
+
畠山 歓、Rumi Nakagawa、takagi、Toshio Nishida、Hideaki Hayashi、p1atdev、Daichi Kohmoto、Nishijima、Chattso-GPT、西井康隆、Jie Zeng、加藤純、堀江吏将、Kazutaka Nishimae、Jinsei Shiraishi、川村 正春、山口 裕輝、Nishi、Esty、Minami Someya、林寛太 (Kanta Hayashi)、菊池満帆、岩田 兼太朗、、江國翔太、佐野敏幸、K. Nishizawa、Susumu Ota、Issei Fujimoto、永原恒治、hiroaki shioya、masaki okamura、y_morinaga、川村 正春、熊田匡仁、Mさん、asaoka_tadashi、Yuki Namiuchi、片上 舜、河越 淳、Issei Fujimoto、Yukie Kawano、Chihiro HIGUCHI、Ryota Mitsuhashi、Chihiro Arata、Atsushi Saito、Kunihiro Watanabe
|
|
|
|
|
|
|
|
|
|