Update README.md
Browse files
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 |
+

|
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 |
+

|
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 |
+
```
|