diff --git a/.github/workflows/ci-llama-large-tests.yaml b/.github/workflows/ci-llama-large-tests.yaml index d79031b8c..53e8535bb 100644 --- a/.github/workflows/ci-llama-large-tests.yaml +++ b/.github/workflows/ci-llama-large-tests.yaml @@ -69,9 +69,7 @@ jobs: -e "git+https://github.com/iree-org/iree-turbine.git#egg=iree-turbine" # Test with pinned nightly releases, not what iree-turbine uses. - pip install -f https://iree.dev/pip-release-links.html --upgrade \ - iree-base-compiler==2.9.0rc20241108 \ - iree-base-runtime==2.9.0rc20241108 \ + pip install --upgrade -r shortfin/requirements-iree-compiler.txt \ "numpy<2.0" - name: Run llama tests diff --git a/.github/workflows/ci-llama-quick-tests.yaml b/.github/workflows/ci-llama-quick-tests.yaml index decd0aa96..9b325dd93 100644 --- a/.github/workflows/ci-llama-quick-tests.yaml +++ b/.github/workflows/ci-llama-quick-tests.yaml @@ -70,9 +70,7 @@ jobs: -e "git+https://github.com/iree-org/iree-turbine.git#egg=iree-turbine" # Test with pinned nightly releases, not what iree-turbine uses. - pip install -f https://iree.dev/pip-release-links.html --upgrade \ - iree-base-compiler==2.9.0rc20241108 \ - iree-base-runtime==2.9.0rc20241108 \ + pip install --upgrade -r shortfin/requirements-iree-compiler.txt \ "numpy<2.0" - name: Run llama 8b tests