Browse Source

Create debug.sh

Allen 1 year ago
parent
commit
636f874001
1 changed files with 4 additions and 0 deletions
  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