diff --git a/llm/android/LlamaDemo/app/src/androidTest/java/com/example/executorchllamademo/PresetSanityTest.kt b/llm/android/LlamaDemo/app/src/androidTest/java/com/example/executorchllamademo/PresetSanityTest.kt index e01fc07a..601d66bc 100644 --- a/llm/android/LlamaDemo/app/src/androidTest/java/com/example/executorchllamademo/PresetSanityTest.kt +++ b/llm/android/LlamaDemo/app/src/androidTest/java/com/example/executorchllamademo/PresetSanityTest.kt @@ -40,7 +40,7 @@ class PresetSanityTest { companion object { private const val TAG = "PresetSanityTest" - private const val DEFAULT_CONFIG_URL = "https://raw.githubusercontent.com/meta-pytorch/executorch-examples/889ccc6e88813cbf03775889beed29b793d0c8db/llm/android/LlamaDemo/app/src/main/assets/preset_models.json" + private const val DEFAULT_CONFIG_URL = "https://raw.githubusercontent.com/meta-pytorch/executorch-examples/615fa601fd75493ab0e1828c14bbd24f83cd5133/llm/android/LlamaDemo/app/src/main/assets/preset_models.json" } @get:Rule diff --git a/llm/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/ui/screens/SelectPresetModelScreen.kt b/llm/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/ui/screens/SelectPresetModelScreen.kt index 67033bcb..c6f9761c 100644 --- a/llm/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/ui/screens/SelectPresetModelScreen.kt +++ b/llm/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/ui/screens/SelectPresetModelScreen.kt @@ -58,7 +58,7 @@ import com.example.executorchllamademo.ui.theme.LocalAppColors import com.example.executorchllamademo.ui.viewmodel.ConfigLoadState import com.example.executorchllamademo.ui.viewmodel.ModelDownloadState -private const val DEFAULT_CONFIG_URL = "https://raw.githubusercontent.com/meta-pytorch/executorch-examples/889ccc6e88813cbf03775889beed29b793d0c8db/llm/android/LlamaDemo/app/src/main/assets/preset_models.json" +private const val DEFAULT_CONFIG_URL = "https://raw.githubusercontent.com/meta-pytorch/executorch-examples/615fa601fd75493ab0e1828c14bbd24f83cd5133/llm/android/LlamaDemo/app/src/main/assets/preset_models.json" @Composable fun SelectPresetModelScreen( diff --git a/llm/android/LlamaDemo/run_instrumentation_test.sh b/llm/android/LlamaDemo/run_instrumentation_test.sh index a699fac8..4915acdd 100644 --- a/llm/android/LlamaDemo/run_instrumentation_test.sh +++ b/llm/android/LlamaDemo/run_instrumentation_test.sh @@ -18,4 +18,5 @@ python -m pytorch_tokenizers.tools.llama2c.convert -t tokenizer.model -o tokeniz adb mkdir -p /data/local/tmp/llama adb push stories110m_h.pte /data/local/tmp/llama adb push tokenizer.bin /data/local/tmp/llama +adb shell sync diff --git a/llm/android/LlamaDemo/scripts/run-ci-tests.sh b/llm/android/LlamaDemo/scripts/run-ci-tests.sh index cfe6f8d6..ffa6049b 100755 --- a/llm/android/LlamaDemo/scripts/run-ci-tests.sh +++ b/llm/android/LlamaDemo/scripts/run-ci-tests.sh @@ -50,6 +50,7 @@ for file in /tmp/llama_models/*; do while [ $retry -lt $max_retries ] && [ "$success" = "false" ]; do # Run push (ignore exit code, verify by checking file on device) timeout $timeout_secs adb push "$file" /data/local/tmp/llama/ || true + adb shell sync # Verify file was pushed by checking it exists and has correct size device_size=$(adb shell "stat -c%s /data/local/tmp/llama/$filename 2>/dev/null || echo 0" | tr -d '\r')