Skip to content

Commit 21f73fd

Browse files
committed
Update on "CI test for lora"
Add CI test for LoRA: note, not program-data separated yet. Differential Revision: [D78751767](https://our.internmc.facebook.com/intern/diff/D78751767/) [ghstack-poisoned]
2 parents 3406c0b + c98e48c commit 21f73fd

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

.ci/scripts/test_llama_lora.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,8 +67,9 @@ $PYTHON_EXECUTABLE -m extension.llm.export.export_llm \
6767
cmake_install_executorch_libraries
6868
cmake_build_llama_runner
6969

70+
PROMPT="What happens if you eat watermelon seeds?"
7071
# Run llama runner
71-
RUNTIME_ARGS="--model_path=${EXPORTED_MODEL_NAME} --tokenizer_path=${DOWNLOADED_PATH}/tokenizer.model --prompt='What happens if you eat watermelon seeds?' --temperature=0 --seq_len=10 --warmup=1"
72+
RUNTIME_ARGS="--model_path=${EXPORTED_MODEL_NAME} --tokenizer_path=${DOWNLOADED_PATH}/tokenizer.model --prompt=${PROMPT} --temperature=0 --seq_len=10 --warmup=1"
7273

7374
NOW=$(date +"%H:%M:%S")
7475
echo "Starting to run llama runner at ${NOW}"

0 commit comments

Comments
 (0)