Aratako commited on
Commit
550d9ae
1 Parent(s): 7f8344f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +103 -0
README.md CHANGED
@@ -1,3 +1,106 @@
1
  ---
 
 
 
2
  license: llama2
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ base_model:
3
+ - tokyotech-llm/Swallow-13b-instruct-hf
4
+ - nitky/Superswallow-13b-v0.2
5
  license: llama2
6
+ language:
7
+ - ja
8
+ tags:
9
+ - mergekit
10
+ - merge
11
+ - MoE
12
  ---
13
+ # Swallow-MoE-2x13B-v0.1
14
+ [**English description here**](#description)
15
+
16
+
17
+ ## 概要
18
+ Llama-2ベースの学習済み日本語モデルである[tokyotech-llm/Swallow-13b-instruct-hf](https://huggingface.co/tokyotech-llm/Swallow-13b-instruct-hf)と、それを利用したマージモデルである[nitky/Superswallow-13b-v0.2](https://huggingface.co/nitky/Superswallow-13b-v0.2)
19
+ を、[mergekit](https://github.com/cg123/mergekit)を使ってMoEを行い作成したモデルです。
20
+
21
+ [GGUF版はこちら](https://huggingface.co/Aratako/Swallow-MoE-2x13B-v0.1-GGUF)
22
+
23
+ 以下2モデルを利用しています。
24
+ - [tokyotech-llm/Swallow-13b-instruct-hf](https://huggingface.co/tokyotech-llm/Swallow-13b-instruct-hf)
25
+ - [nitky/Superswallow-13b-v0.2](https://huggingface.co/nitky/Superswallow-13b-v0.2)
26
+
27
+ ## ライセンス
28
+ Llama2ライセンスを継承しますが、それ以外にもこのマージモデルはLlama2ライセンス以外のライセンスの影響を受けます。
29
+ - [nitky/Superswallow-13b-v0.2](https://huggingface.co/nitky/Superswallow-13b-v0.2)の利用により。[AI2 ImpACT license](https://allenai.org/impact-license)を継承している可能性があります。詳細は元モデルの概要をご確認ください。
30
+
31
+ ## ベンチマーク
32
+ ベースとしたSwallow-13b-instruct-hf、Superswallow-13b-v0.2と本モデルの[japanese-mt-bench](https://github.com/Stability-AI/FastChat/tree/jp-stable/fastchat/llm_judge)の結果は以下の通りです。
33
+ (シングルターン)
34
+ |Model|Size|Coding|Extraction|Humanities|Math|Reasoning|Roleplay|STEM|Writing|avg_score|
35
+ |---|---|---|---|---|---|---|---|---|---|---|
36
+ | Swallow-13b-instruct-hf | 13B | 2.1 | 5.0 | 6.1 | 1.9 | 4.5 | 5.0 | 4.9 | 5.6 | 4.3875 |
37
+ | Superswallow-13b-v0.2 | 13B | **2.7** | **6.3** | **8.4** | 2.2 | **5.9** | 6.8 | 7.2 | 6.8 | 5.7875 |
38
+ | This model | 2x13B | 2.6 | 6.2 | **8.4** | **2.3** | 5.3 | **7.4** | **7.7** | **8.4** | **6.0375** |
39
+
40
+ ![レーダーチャート](./japanese_mt_bench.png)
41
+ **ベンチマークに使用したプロンプト**
42
+ - Swallow-13b-instruct-hf, Superswallow-13b-v0.2
43
+ ```
44
+ 以下に、あるタスクを説明する指示があり、それに付随する入力が更なる文脈を提供しています。リクエストを適切に完了するための回答を記述してください。
45
+ ### 指示:
46
+ {instruction}
47
+ ### 応答:
48
+ ```
49
+ - Swallow-MoE-2x13B-v0.1
50
+ ```
51
+ ### 指示:
52
+ {instruction}
53
+ ### 応答:
54
+ ```
55
+
56
+ ## Description
57
+ This model is created using MoE (Mixture of Experts) through mergekit based on [tokyotech-llm/Swallow-13b-instruct-hf](https://huggingface.co/tokyotech-llm/Swallow-13b-instruct-hf) and [nitky/Superswallow-13b-v0.2](https://huggingface.co/nitky/Superswallow-13b-v0.2).
58
+ [Click here for the GGUF version](https://huggingface.co/Aratako/Swallow-MoE-2x13B-v0.1-GGUF)
59
+ It utilizes the following two models:
60
+ - [tokyotech-llm/Swallow-13b-instruct-hf](https://huggingface.co/tokyotech-llm/Swallow-13b-instruct-hf)
61
+ - [nitky/Superswallow-13b-v0.2](https://huggingface.co/nitky/Superswallow-13b-v0.2)
62
+ -
63
+ ## License
64
+ While inheriting the Llama2 license, this merged model is also subject to other licenses due to its use of models beyond Llama2.
65
+ - Due to the use of [nitky/Superswallow-13b-v0.2](https://huggingface.co/nitky/Superswallow-13b-v0.2), it may inherit the [AI2 ImpACT license](https://allenai.org/impact-license). Please refer to the original model's overview for details.
66
+
67
+ ## Benchmark
68
+ The results of this model and the base Swallow-13b-instruct-hf, Superswallow-13b-v0.2 on japanese-mt-bench are as follows.
69
+ (Single turn)
70
+ |Model|Size|Coding|Extraction|Humanities|Math|Reasoning|Roleplay|STEM|Writing|avg_score|
71
+ |---|---|---|---|---|---|---|---|---|---|---|
72
+ | Swallow-13b-instruct-hf | 13B | 2.1 | 5.0 | 6.1 | 1.9 | 4.5 | 5.0 | 4.9 | 5.6 | 4.3875 |
73
+ | Superswallow-13b-v0.2 | 13B | **2.7** | **6.3** | **8.4** | 2.2 | **5.9** | 6.8 | 7.2 | 6.8 | 5.7875 |
74
+ | This model | 2x13B | 2.6 | 6.2 | **8.4** | **2.3** | 5.3 | **7.4** | **7.7** | **8.4** | **6.0375** |
75
+
76
+ ![レーダーチャート](./japanese_mt_bench.png)
77
+
78
+ **Prompt used for benchmark**
79
+ ```
80
+ - Swallow-13b-instruct-hf, Superswallow-13b-v0.2
81
+ ```
82
+ 以下に、あるタスクを説明する指示があり、それに付随する入力が更なる文脈を提供しています。リクエストを適切に完了するための回答を記述してください。
83
+ ### 指示:
84
+ {instruction}
85
+ ### 応答:
86
+ ```
87
+ - Swallow-MoE-2x13B-v0.1
88
+ ```
89
+ ### 指示:
90
+ {instruction}
91
+ ### 応答:
92
+ ```
93
+
94
+ ## Merge config
95
+ [mergekit_config.yml](./mergekit_moe_config.yml)
96
+ ```yaml
97
+ base_model: ./Superswallow-13b-v0.2
98
+ gate_mode: random
99
+ dtype: bfloat16
100
+ experts:
101
+ - source_model: ./Superswallow-13b-v0.2
102
+ positive_prompts: []
103
+ - source_model: ./Swallow-13b-instruct-hf
104
+ positive_prompts: []
105
+ tokenizer_source: union
106
+ ```