diff --git a/recipes/A5000_24GB_x8/summarizer-en-cnn-dailymail.yaml b/recipes/A5000_24GB_x8/summarizer-en-cnn-dailymail.yaml index 867ace2..58890ae 100644 --- 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