exp.sh 623 B

12345678910111213141516
  1. # CUDA_VISIBLE_DEVICES=$1 python -u generation.py \
  2. # --input-path data/summarization/xsum.jsonl \
  3. # --output-path summarization_output/xsum_baseline.jsonl \
  4. # --model-name meta-llama/Llama-2-7b-hf
  5. # 20.46/4.9/15.11
  6. CUDA_VISIBLE_DEVICES=$1 python -u generation.py \
  7. --input-path data/summarization/xsum.jsonl \
  8. --output-path summarization_output/xsum_h2o.jsonl \
  9. --model-name meta-llama/Llama-2-7b-hf \
  10. --enable_h2o_generation
  11. # CUDA_VISIBLE_DEVICES=$1 python -u generation.py \
  12. # --input-path data/summarization/xsum.jsonl \
  13. # --output-path summarization_output/xsum_h2o.jsonl \
  14. # --model-name meta-llama/Llama-2-7b-hf