# CUDA_VISIBLE_DEVICES=$1 python -u generation.py \
# --input-path data/summarization/xsum.jsonl \
# --output-path summarization_output/xsum_baseline.jsonl \
# --model-name meta-llama/Llama-2-7b-hf 
# 20.46/4.9/15.11

CUDA_VISIBLE_DEVICES=$1 python -u generation.py \
--input-path data/summarization/xsum.jsonl \
--output-path summarization_output/xsum_h2o.jsonl \
--model-name meta-llama/Llama-2-7b-hf \
--enable_h2o_generation