LLamaRecipes/research/long-context-llama/H2O/exp.sh

# 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 

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