Update README.md
Browse files
README.md
CHANGED
@@ -1,42 +1,53 @@
|
|
1 |
-
---
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
-
|
6 |
-
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
|
26 |
-
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
42 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
datasets:
|
4 |
+
- Aratako/Rosebleu-1on1-Dialogues-RP
|
5 |
+
- Aratako/LimaRP-augmented-ja-karakuri
|
6 |
+
- grimulkan/LimaRP-augmented
|
7 |
+
- Aratako/Bluemoon_Top50MB_Sorted_Fixed_ja
|
8 |
+
- SicariusSicariiStuff/Bluemoon_Top50MB_Sorted_Fixed
|
9 |
+
- OmniAICreator/Japanese-Roleplay
|
10 |
+
language:
|
11 |
+
- ja
|
12 |
+
library_name: transformers
|
13 |
+
tags:
|
14 |
+
- roleplay
|
15 |
+
base_model:
|
16 |
+
- Aratako/Ninja-v1-RP-WIP
|
17 |
+
---
|
18 |
+
|
19 |
+
# Ninja-v1-RP
|
20 |
+
|
21 |
+
## 概要
|
22 |
+
This is a merge of pre-trained language models created using [mergekit](https://github.com/cg123/mergekit).
|
23 |
+
|
24 |
+
[Aratako/Ninja-v1-RP-WIP](https://huggingface.co/Aratako/Ninja-v1-RP-WIP)をベースに、Task Vectorの加算・Model Stockによるマージを行い指示追従能力と表現力を強化したロールプレイ用モデルです。
|
25 |
+
|
26 |
+
## マージの詳細
|
27 |
+
まず、[Aratako/Ninja-v1-RP-WIP](https://huggingface.co/Aratako/Ninja-v1-RP-WIP)に対し、以下4モデルのTask Vectorを0.8倍して加算し、4種類、Task Vector加算モデルを作成しました。
|
28 |
+
|
29 |
+
- [senseable/WestLake-7B-v2](https://huggingface.co/senseable/WestLake-7B-v2)
|
30 |
+
- [SanjiWatsuki/Kunoichi-DPO-v2-7B](https://huggingface.co/SanjiWatsuki/Kunoichi-DPO-v2-7B)
|
31 |
+
- [SanjiWatsuki/Silicon-Maid-7B](https://huggingface.co/SanjiWatsuki/Silicon-Maid-7B)
|
32 |
+
- [SanjiWatsuki/Loyal-Macaroni-Maid-7B](https://huggingface.co/SanjiWatsuki/Loyal-Macaroni-Maid-7B)
|
33 |
+
|
34 |
+
各モデルのTask Vectorの加算の式は以下の通りです。
|
35 |
+
```
|
36 |
+
new_model = Ninja-v1-RP-WIP + 0.8 * (target_model - Mistral-7B-v0.1)
|
37 |
+
```
|
38 |
+
|
39 |
+
次に、このTask Vector加算によってできた4モデルと元のモデルを、Model Stockという手法を用い以下のようなconfigを使ってmergekitでマージし、このモデルを作成しました。
|
40 |
+
|
41 |
+
```yaml
|
42 |
+
models:
|
43 |
+
- model: ./Ninja-v1-RP-WIP
|
44 |
+
- model: ./Ninja-v1-RP-WIP-Kunoichi
|
45 |
+
- model: ./Ninja-v1-RP-WIP-SiliconMaid
|
46 |
+
- model: ./Ninja-v1-RP-WIP-WestLake
|
47 |
+
- model: ./Ninja-v1-RP-WIP-LoyalMacaroniMaid
|
48 |
+
merge_method: model_stock
|
49 |
+
base_model: ./Ninja-v1-RP-WIP
|
50 |
+
dtype: bfloat16
|
51 |
+
tokenizer_source: union
|
52 |
+
|
53 |
+
```
|