فهرست منبع

Update exp.sh

Allen 1 سال پیش
والد
کامیت
532e88bd41
1فایلهای تغییر یافته به همراه7 افزوده شده و 7 حذف شده
  1. 7 7
      research/long-context-llama/H2O/exp.sh

+ 7 - 7
research/long-context-llama/H2O/exp.sh

@@ -4,10 +4,10 @@ CUDA_VISIBLE_DEVICES=$1 python -u generate.py \
 --model-name meta-llama/Llama-2-7b-hf 
 
 
-CUDA_VISIBLE_DEVICES=$1 python -u generate.py \
---input-path data/summarization/xsum.jsonl \
---output-path xsum_h2o.jsonl \
---model-name meta-llama/Llama-2-7b-hf \
---enable_h2o_generation \
---num_heavy_hitter_tokens 256 \
---num_local_windows 256
+# CUDA_VISIBLE_DEVICES=$1 python -u generate.py \
+# --input-path data/summarization/xsum.jsonl \
+# --output-path xsum_h2o.jsonl \
+# --model-name meta-llama/Llama-2-7b-hf \
+# --enable_h2o_generation \
+# --num_heavy_hitter_tokens 256 \
+# --num_local_windows 256