|
@@ -1,12 +1,12 @@
|
|
|
CUDA_VISIBLE_DEVICES=$1 python -u generation.py \
|
|
|
--input-path data/summarization/xsum.jsonl \
|
|
|
---output-path xsum_baseline.jsonl \
|
|
|
+--output-path summarization_output/xsum_baseline.jsonl \
|
|
|
--model-name meta-llama/Llama-2-7b-hf
|
|
|
|
|
|
|
|
|
# CUDA_VISIBLE_DEVICES=$1 python -u generation.py \
|
|
|
# --input-path data/summarization/xsum.jsonl \
|
|
|
-# --output-path xsum_h2o.jsonl \
|
|
|
+# --output-path summarization_output/xsum_h2o.jsonl \
|
|
|
# --model-name meta-llama/Llama-2-7b-hf \
|
|
|
# --enable_h2o_generation \
|
|
|
# --num_heavy_hitter_tokens 256 \
|