diff --git a/.ci/scripts/setup-linux.sh b/.ci/scripts/setup-linux.sh index 89a7f3b4985..f35bfa5a0db 100755 --- a/.ci/scripts/setup-linux.sh +++ b/.ci/scripts/setup-linux.sh @@ -23,4 +23,3 @@ fi # of nightly. This allows CI to test against latest commits from PyTorch install_executorch "use-pt-pinned-commit" build_executorch_runner "${BUILD_TOOL}" -do_not_use_nightly_on_ci diff --git a/pyproject.toml b/pyproject.toml index 11673f9c960..65da50c03e6 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -65,6 +65,9 @@ dependencies=[ "ruamel.yaml", "sympy", "tabulate", + "torch==2.6.0", + "torchaudio==2.6.0", + "torchvision==0.21.0", "typing-extensions", ]