python ./code/run_summarization.py \
    --model_name_or_path google/mt5-small \
    --train_file train.jsonl \
    --validation_file public.jsonl \
    --dataset_config "3.0.0" \
    --source_prefix "summarize: " \
    --output_dir ./HW2 \
    --num_train_epochs 3 \
    --num_beams 1 \
    --do_sample False \
    --push_to_hub

bash ./download.sh
bash ./run.sh /path/to/input.jsonl /path/to/output.jsonl