@@ -96,15 +96,15 @@ test_model() {
96
96
bash examples/models/llama/install_requirements.sh
97
97
# Test export_llama script: python3 -m examples.models.llama.export_llama.
98
98
# Use Llama random checkpoint with Qwen 2.5 1.5b model configuration.
99
- " ${PYTHON_EXECUTABLE} " -m examples.models.llama.export_llama --model " ${MODEL_NAME} " -c examples/models/llama/params/demo_rand_params.pth - p examples/models/qwen2_5/1_5b_config.json
99
+ " ${PYTHON_EXECUTABLE} " -m examples.models.llama.export_llama --model " ${MODEL_NAME} " -p examples/models/qwen2_5/1_5b_config.json
100
100
rm " ./${MODEL_NAME} .pte"
101
101
return # Skip running with portable executor runnner since portable doesn't support Qwen's biased linears.
102
102
fi
103
- if [[ " ${MODEL_NAME} " == " phi-4-mini " ]]; then
103
+ if [[ " ${MODEL_NAME} " == " phi_4_mini " ]]; then
104
104
# Install requirements for export_llama
105
105
bash examples/models/llama/install_requirements.sh
106
106
# Test export_llama script: python3 -m examples.models.llama.export_llama.
107
- " ${PYTHON_EXECUTABLE} " -m examples.models.llama.export_llama --model " ${MODEL_NAME} " -c examples/models/llama/params/demo_rand_params.pth - p examples/models/phi-4-mini /config.json
107
+ " ${PYTHON_EXECUTABLE} " -m examples.models.llama.export_llama --model " ${MODEL_NAME} " -p examples/models/phi_4_mini /config.json
108
108
run_portable_executor_runner
109
109
rm " ./${MODEL_NAME} .pte"
110
110
return
@@ -224,19 +224,22 @@ test_model_with_coreml() {
224
224
225
225
" ${PYTHON_EXECUTABLE} " -m examples.apple.coreml.scripts.export --model_name=" ${MODEL_NAME} " --compute_precision " ${DTYPE} "
226
226
EXPORTED_MODEL=$( find " ." -type f -name " ${MODEL_NAME} *.pte" -print -quit)
227
- # TODO:
227
+
228
228
if [ -n " $EXPORTED_MODEL " ]; then
229
229
EXPORTED_MODEL_WITH_DTYPE=" ${EXPORTED_MODEL% .pte} _${DTYPE} .pte"
230
230
mv " $EXPORTED_MODEL " " $EXPORTED_MODEL_WITH_DTYPE "
231
231
EXPORTED_MODEL=" $EXPORTED_MODEL_WITH_DTYPE "
232
- echo " Renamed file path : $EXPORTED_MODEL "
232
+ echo " OK exported model : $EXPORTED_MODEL "
233
233
else
234
- echo " No .pte file found"
234
+ echo " [error] failed to export model: no .pte file found"
235
235
exit 1
236
236
fi
237
237
238
238
# Run the model
239
239
if [ " ${should_test} " = true ]; then
240
+ echo " Installing requirements needed to build coreml_executor_runner..."
241
+ backends/apple/coreml/scripts/install_requirements.sh
242
+
240
243
echo " Testing exported model with coreml_executor_runner..."
241
244
local out_dir=$( mktemp -d)
242
245
COREML_EXECUTOR_RUNNER_OUT_DIR=" ${out_dir} " examples/apple/coreml/scripts/build_executor_runner.sh
0 commit comments