- # 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
|