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

Author: yuiseki

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