瀏覽代碼

Create debug.sh

Allen 1 年之前
父節點
當前提交
636f874001
共有 1 個文件被更改,包括 4 次插入0 次删除
  1. 4 0
      recipes/experimental/long-context/H2O/src/debug.sh

+ 4 - 0
recipes/experimental/long-context/H2O/src/debug.sh

@@ -0,0 +1,4 @@
+CUDA_VISIBLE_DEVICES=1 python -u run_needle_haystack_test.py \
+--input-path data/needle_test/Huggingface \
+--output-path needle_test_results/huggingface/llama-3-8b-instruct/ \
+--model-name meta-llama/Meta-Llama-3-8B-Instruct