Update README.md
Browse files
README.md
CHANGED
@@ -8,6 +8,33 @@ pipeline_tag: text-generation
|
|
8 |
inference: false
|
9 |
license: apache-2.0
|
10 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
## Training procedure
|
12 |
|
13 |
Finetune [tiiuae/falcon-7b](https://huggingface.co/tiiuae/falcon-7b) with [ohtaman/kokkai2022](https://huggingface.co/datasets/ohtaman/kokkai2022)(currentry, private) dataset with LoRA.
|
@@ -24,7 +51,7 @@ The training parameters are
|
|
24 |
|optimizer|AdamW|
|
25 |
|lr|4e-4|
|
26 |
|
27 |
-
the prompt is
|
28 |
|
29 |
```
|
30 |
# question
|
@@ -39,10 +66,6 @@ the prompt is something like
|
|
39 |
|
40 |
```
|
41 |
|
42 |
-
### Framework versions
|
43 |
-
|
44 |
-
- PEFT 0.4.0.dev0
|
45 |
-
|
46 |
### Example Notebook (Colab)
|
47 |
|
48 |
[Colaboratory](https://colab.research.google.com/drive/1oWHM5_DbltvrD27oZL4-fumXChkMkrC5?usp=sharing) (Pro is not needed.)
|
|
|
8 |
inference: false
|
9 |
license: apache-2.0
|
10 |
---
|
11 |
+
|
12 |
+
<a href="https://colab.research.google.com/drive/1oWHM5_DbltvrD27oZL4-fumXChkMkrC5?usp=sharing">
|
13 |
+
<img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab">
|
14 |
+
</a>
|
15 |
+
|
16 |
+
This model learned the proceedings of the Japanese parliament in 2022.
|
17 |
+
The [dataset](https://huggingface.co/datasets/ohtaman/kokkai2022) is collected using
|
18 |
+
[National Diet Library's Search API](https://kokkai.ndl.go.jp/api.html).
|
19 |
+
|
20 |
+
example input:
|
21 |
+
|
22 |
+
```
|
23 |
+
# question
|
24 |
+
麻生太郎
|
25 |
+
|
26 |
+
増税が必要とお考えでしょうか?
|
27 |
+
# answer
|
28 |
+
鈴木 俊一
|
29 |
+
```
|
30 |
+
|
31 |
+
output:
|
32 |
+
|
33 |
+
```
|
34 |
+
「財政民主主義」のためには、国庫負担を引き下げるならば、企業の賃上げを実現するためには、消費者物価高対策を行っていく。
|
35 |
+
今回の補正予算案では、二・五兆円の経済対策を盛り込んでおります。特に、小売店等に向けた新規就業支援事業、零細化対策、マイクロプラットフォーム構築事業、地域活性化事業、公共施設整備事業、交通基盤整備事業、都道府県単位の自動車検査事業、地方創生臨時交付金、地方創生臨時交付金、地方創生臨時交付金、地方創生臨時交付金、地方創生臨時交付金、地方創生臨時交付金、地方創生臨時交付金、地方創生臨時交付金、地方創生臨時交付金、地方創生臨時交付金、地方創生臨時交付金、地方創生臨時交付金、地方創生臨時交付金、地方創生臨時交付金、地方創生臨時交付金、地方創生臨時交付金、
|
36 |
+
```
|
37 |
+
|
38 |
## Training procedure
|
39 |
|
40 |
Finetune [tiiuae/falcon-7b](https://huggingface.co/tiiuae/falcon-7b) with [ohtaman/kokkai2022](https://huggingface.co/datasets/ohtaman/kokkai2022)(currentry, private) dataset with LoRA.
|
|
|
51 |
|optimizer|AdamW|
|
52 |
|lr|4e-4|
|
53 |
|
54 |
+
the prompt template is as follows:
|
55 |
|
56 |
```
|
57 |
# question
|
|
|
66 |
|
67 |
```
|
68 |
|
|
|
|
|
|
|
|
|
69 |
### Example Notebook (Colab)
|
70 |
|
71 |
[Colaboratory](https://colab.research.google.com/drive/1oWHM5_DbltvrD27oZL4-fumXChkMkrC5?usp=sharing) (Pro is not needed.)
|