Skip to content

Commit a49e08b

Browse files
committed
Fix model name
1 parent d5cb6b3 commit a49e08b

File tree

2 files changed

+31
-1
lines changed

2 files changed

+31
-1
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
target_task: tasks/i18n/ja.md
2+
base_model_id: TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T
3+
model_name: tinyllama-summarizer-en-cnn-dailymail-3T-v0.1
4+
output_base_dir: /data/output
5+
dataset_id: cnn_dailymail
6+
dataset_input_field_name: instruction
7+
dataset_context_field_name: input
8+
dataset_output_field_name: output
9+
dataset_train_split_seed: 42
10+
dataset_train_split_test_size: 0.2
11+
lora_r: 8
12+
lora_alpha: 16
13+
lora_dropout: 0.05
14+
train_claim_gpu_num: 3
15+
train_per_device_train_batch_size: 16
16+
train_gradient_accumulation_steps: 4
17+
train_num_train_epochs: 4
18+
train_max_steps: 1000
19+
train_fp16: True
20+
inference_max_new_tokens: 32
21+
evaluations:
22+
-
23+
prompt: "火縄銃の威力が全国に知られる事となった、1575年に織田・徳川連合軍が鉄砲隊を用いて武田勝頼率いる騎馬隊を破った戦いを何というでしょう?"
24+
expected_output: "長篠の戦いです。"
25+
-
26+
prompt: "ベトナム戦争終結や米ソ戦略兵器削減交渉などを進めたものの、1974年にウォーターゲート事件の責任をとって辞任したアメリカの第37代大統領は誰でしょう?"
27+
expected_output: "リチャード・ニクソンです。"
28+
-
29+
prompt: "格闘家ボブ・サップの出身国はどこでしょう?"
30+
expected_output: "アメリカです。"

recipes/A5000_24GB_x8/summarizer-ja-cnn-dailymail.yaml

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
target_task: tasks/i18n/ja.md
22
base_model_id: yuiseki/tinyllama-ja-wikipedia-1.5T-v0.1
3-
model_name: tinyllama-ja-wikipedia-summarizer-cnn-dailymail-1.5T-v0.1
3+
model_name: tinyllama-summarizer-ja-wikipedia-cnn-dailymail-1.5T-v0.1
44
output_base_dir: /data/output
55
dataset_id: kunishou/cnn-dailymail-27k-ja
66
dataset_input_field_name: instruction

0 commit comments

Comments
 (0)