Update finetune_sample/lora_test.py
Browse files
finetune_sample/lora_test.py
CHANGED
|
@@ -1,3 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
| 1 |
from transformers import AutoTokenizer
|
| 2 |
from auto_gptq import AutoGPTQForCausalLM
|
| 3 |
from peft import PeftModel, PeftConfig
|
|
@@ -30,7 +33,7 @@ prompt_text = [
|
|
| 30 |
"劇場版アニメ『風の谷のナウシカ』はいつ公開されましたか？",
|
| 31 |
"漫画『風の谷のナウシカ』全巻の累計発行部数は何部ですか？",
|
| 32 |
"『風の谷のナウシカ』は何年に歌舞伎化されましたか？",
|
| 33 |
-
|
| 34 |
"『風の谷のナウシカ』に出演するクロトワはどの大学を卒業していますか？"]
|
| 35 |
|
| 36 |
for j in range(len(prompt_text)):
|
|
|
|
| 1 |
+
# There is a blog post explaining the results. (Japanese)
|
| 2 |
+
# https://webbigdata.jp/post-20104/
|
| 3 |
+
|
| 4 |
from transformers import AutoTokenizer
|
| 5 |
from auto_gptq import AutoGPTQForCausalLM
|
| 6 |
from peft import PeftModel, PeftConfig
|
|
|
|
| 33 |
"劇場版アニメ『風の谷のナウシカ』はいつ公開されましたか？",
|
| 34 |
"漫画『風の谷のナウシカ』全巻の累計発行部数は何部ですか？",
|
| 35 |
"『風の谷のナウシカ』は何年に歌舞伎化されましたか？",
|
| 36 |
+
"『風の谷のナウシカ』で、ナウシカが主に乗る飛行メカは何ですか？",
|
| 37 |
"『風の谷のナウシカ』に出演するクロトワはどの大学を卒業していますか？"]
|
| 38 |
|
| 39 |
for j in range(len(prompt_text)):
|