ohtaman commited on
Commit
a9f07fc
1 Parent(s): 5ee3d4c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +28 -5
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 something like
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.)