From bc65c6704c2c1fd706d71f7f86be4bc81eee8f57 Mon Sep 17 00:00:00 2001 From: Jack Zhang Date: Wed, 13 Nov 2024 11:16:10 -0800 Subject: [PATCH] Fix broken trunk test_model.sh --- .ci/scripts/test_model.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.ci/scripts/test_model.sh b/.ci/scripts/test_model.sh index 4e37d0ebaa3..cd9583ff97e 100755 --- a/.ci/scripts/test_model.sh +++ b/.ci/scripts/test_model.sh @@ -77,7 +77,7 @@ test_model() { # Install requirements for export_llama bash examples/models/llama/install_requirements.sh # Test export_llama script: python3 -m examples.models.llama.export_llama - "${PYTHON_EXECUTABLE}" -m examples.models.llama.export_llama -c examples/models/llama/params/demo_rand_params.pth -p examples/models/llama/params/demo_config.json + "${PYTHON_EXECUTABLE}" -m examples.models.llama.export_llama --model "${MODEL_NAME}" -c examples/models/llama/params/demo_rand_params.pth -p examples/models/llama/params/demo_config.json run_portable_executor_runner rm "./${MODEL_NAME}.pte" fi