Use stories15m

Author: kirklandsign

extension/android/executorch_android/android_test_setup.sh

@@ -16,14 +16,14 @@ BASEDIR=$(dirname "$(realpath $0)")
 cp "${BASEDIR}/../../../extension/module/test/resources/add.pte" "${BASEDIR}/src/androidTest/resources"
 
 pushd "${BASEDIR}/../../../"
-curl -C - -Ls "https://huggingface.co/karpathy/tinyllamas/resolve/main/stories110M.pt" --output stories110M.pt
+curl -C - -Ls "https://huggingface.co/karpathy/tinyllamas/resolve/main/stories15M.pt" --output stories15M.pt
 curl -C - -Ls "https://raw.githubusercontent.com/karpathy/llama2.c/master/tokenizer.model" --output tokenizer.model
 # Create params.json file
 touch params.json
-echo '{"dim": 768, "multiple_of": 32, "n_heads": 12, "n_layers": 12, "norm_eps": 1e-05, "vocab_size": 32000}' > params.json
-python -m examples.models.llama.export_llama -c stories110M.pt -p params.json -d fp16 -n stories110m_h.pte -kv
+echo '{"dim": 288, "multiple_of": 32, "n_heads": 6, "n_layers": 6, "norm_eps": 1e-05, "vocab_size": 32000}' > params.json
+python -m examples.models.llama.export_llama -c stories15M.pt -p params.json -d fp16 -n stories15m_h.pte -kv
 python -m pytorch_tokenizers.tools.llama2c.convert -t tokenizer.model -o tokenizer.bin
 
-cp stories110m_h.pte "${BASEDIR}/src/androidTest/resources/stories.pte"
+cp stories15m_h.pte "${BASEDIR}/src/androidTest/resources/stories.pte"
 cp tokenizer.bin "${BASEDIR}/src/androidTest/resources/tokenizer.bin"
 popd

scripts/run_android_emulator.sh

@@ -22,8 +22,7 @@ adb uninstall org.pytorch.executorch.test || true
 adb install -t android-test-debug-androidTest.apk
 
 adb logcat -c
-adb shell am instrument -w -r -e class org.pytorch.executorch.ModuleInstrumentationTest \
-  org.pytorch.executorch.test/androidx.test.runner.AndroidJUnitRunner >result.txt 2>&1
+adb shell am instrument -w -r -e class org.pytorch.executorch.ModuleInstrumentationTest >result.txt 2>&1
 adb logcat -d > logcat.txt
 cat logcat.txt
 grep -q FAILURES result.txt && cat result.txt