Parcourir la source

Update exp.sh

Allen il y a 1 an
Parent
commit
da6f5ebf9b
1 fichiers modifiés avec 2 ajouts et 2 suppressions
  1. 2 2
      research/long-context-llama/H2O/exp.sh

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

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