diff --git a/.gitignore b/.gitignore index 553729e9b68..08d14e13582 100644 --- a/.gitignore +++ b/.gitignore @@ -26,6 +26,7 @@ pip-out/ *.model tokenizer.json *.pte +*.ptd !test_bpe_tokenizer.bin !test_tiktoken_tokenizer.model diff --git a/extension/android/executorch_android/android_test_setup.sh b/extension/android/executorch_android/android_test_setup.sh index f521dac30c5..7ca567c281c 100644 --- a/extension/android/executorch_android/android_test_setup.sh +++ b/extension/android/executorch_android/android_test_setup.sh @@ -18,6 +18,12 @@ prepare_add() { python3 -m test.models.export_program --modules "ModuleAdd" --outdir "${BASEDIR}/src/androidTest/resources/" } +prepare_xor() { + python3 -m extension.training.examples.XOR.export_model --outdir "${BASEDIR}/src/androidTest/resources/" + mv "${BASEDIR}/src/androidTest/resources/xor.pte" "${BASEDIR}/src/androidTest/resources/xor_only.pte" + python3 -m extension.training.examples.XOR.export_model --outdir "${BASEDIR}/src/androidTest/resources/" --external +} + prepare_tinyllama() { pushd "${BASEDIR}/../../../" curl -C - -Ls "https://huggingface.co/karpathy/tinyllamas/resolve/main/stories15M.pt" --output stories15M.pt @@ -43,5 +49,6 @@ prepare_vision() { } prepare_add +prepare_xor prepare_tinyllama prepare_vision