Skip to content

Commit 304855b

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 21f73fd + b2e9a56 commit 304855b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

.ci/scripts/test_llama_lora.sh

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -67,9 +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?"
70+
PROMPT="\"What happens if you eat watermelon seeds?\""
7171
# Run llama runner
72-
RUNTIME_ARGS="--model_path=${EXPORTED_MODEL_NAME} --tokenizer_path=${DOWNLOADED_PATH}/tokenizer.model --prompt=${PROMPT} --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=20 --warmup=1"
7373

7474
NOW=$(date +"%H:%M:%S")
7575
echo "Starting to run llama runner at ${NOW}"
@@ -79,7 +79,7 @@ NOW=$(date +"%H:%M:%S")
7979
echo "Finished at ${NOW}"
8080

8181
RESULT=$(cat result.txt)
82-
EXPECTED_PREFIX="What happens if you eat watermelon seeds? Eating watermelon seeds can be a bit tricky,"
82+
EXPECTED_PREFIX="What happens if you eat watermelon seeds? Watermelon seeds are a good source of vitamin C,"
8383

8484
if [[ "${RESULT}" == "${EXPECTED_PREFIX}"* ]]; then
8585
echo "Expected result prefix: ${EXPECTED_PREFIX}"

0 commit comments

Comments
 (0)