kanhatakeyama commited on
Commit
adc8eaa
1 Parent(s): 12e11bb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +44 -29
README.md CHANGED
@@ -12,16 +12,14 @@ tags:
12
 
13
  ## モデルについて
14
 
15
- Tanuki-8Bは、フルスクラッチでトークン事前学習された8Bパラメータの大規模言語モデルです。
16
- Tanuki-8B-dpo-v1.0は、Tanuki-8BにSFTおよびDPOを適用し作成された対話用モデルです。
17
-
 
 
18
 
19
  ## 量子化モデル
20
- - [AWQ 4bit量子化](https://huggingface.co/team-hatakeyama-phase2/Tanuki-8B-dpo-v1.0-AWQ)
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": "AIによって私たちの暮らしはどのように変わりますか?"}
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
- <s>以下は、タスクを説明する指示です。要求を適切に満たす応答を書きなさい。
52
-
53
- ### 指示:
54
- AIによって私たちの暮らしはどのように変わりますか?
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
55
 
56
- ### 応答:
57
-
58
- ```
59
 
60
- なお、本モデルはデフォルトのシステムプロンプトである`以下は、タスクを説明する指示です。要求を適切に満たす応答を書きなさい。`以外を学習していないため、このシステムプロンプトの使用を推奨します。タスクの詳細はユーザープロンプトに記述してください。
 
 
 
61
 
62
- ## ベンチマーク
63
 
64
- **Japanese MT-Bench**
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
- GPT-4oによる評価 (Nejumi Leaderboard 3)
80
- ![image/png](https://cdn-uploads.huggingface.co/production/uploads/6348501e50fe0799927c3667/TMl4VIoTyUzzdbj5PdVEi.png)
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