Update recipes/A5000_24GB_x8/summarizer-en-cnn-dailymail.yaml

yuiseki · Mar 28, 2024 · 42a4a17 · 42a4a17
1 parent a49e08b
commit 42a4a17
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/recipes/A5000_24GB_x8/summarizer-en-cnn-dailymail.yaml b/recipes/A5000_24GB_x8/summarizer-en-cnn-dailymail.yaml
@@ -3,9 +3,10 @@ base_model_id: TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T
 model_name: tinyllama-summarizer-en-cnn-dailymail-3T-v0.1
 output_base_dir: /data/output
 dataset_id: cnn_dailymail
-dataset_input_field_name: instruction
-dataset_context_field_name: input
-dataset_output_field_name: output
+dataset_load_config: "3.0.0"
+dataset_input_hint: Briefly summarize the following text.
+dataset_input_field_name: article
+dataset_output_field_name: highlights
 dataset_train_split_seed: 42
 dataset_train_split_test_size: 0.2
 lora_r: 8