diff --git a/.ci/scripts/test_model.sh b/.ci/scripts/test_model.sh index c056b5ac98a..ce6ab58d79d 100755 --- a/.ci/scripts/test_model.sh +++ b/.ci/scripts/test_model.sh @@ -48,6 +48,16 @@ prepare_artifacts_upload() { fi } +install_model_requirements() { + if [[ "${MODEL_NAME}" == "yolo26" ]]; then + local torch_url="https://download.pytorch.org/whl/cpu" + "${PYTHON_EXECUTABLE}" -m pip install \ + --upgrade-strategy only-if-needed \ + --extra-index-url "${torch_url}" \ + -r examples/models/yolo26/requirements.txt + fi +} + build_cmake_executor_runner() { local backend_string_select="${1:-}" @@ -339,6 +349,8 @@ test_model_with_cuda() { } +install_model_requirements + if [[ "${BACKEND}" == "portable" ]]; then echo "Testing ${MODEL_NAME} with portable kernels..." test_model