소스 검색

Update exp.sh

Allen 1 년 전
부모
커밋
71d579e112
1개의 변경된 파일1개의 추가작업 그리고 3개의 파일을 삭제
  1. 1 3
      research/long-context-llama/H2O/exp.sh

+ 1 - 3
research/long-context-llama/H2O/exp.sh

@@ -8,6 +8,4 @@ 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 \
---num_heavy_hitter_tokens 256 \
---num_local_windows 256
+--enable_h2o_generation