4949# Since ExecuTorch often uses main-branch features of pytorch, only the nightly
5050# pip versions will have the required features. The NIGHTLY_VERSION value should
5151# agree with the third-party/pytorch pinned submodule commit.
52- #
53- # NOTE: If a newly-fetched version of the executorch repo changes the value of
54- # NIGHTLY_VERSION, you should re-run this script to install the necessary
55- # package versions.
56- NIGHTLY_VERSION=dev20240324
5752
5853# The pip repository that hosts nightly torch packages.
59- TORCH_NIGHTLY_URL =" https://download.pytorch.org/whl/nightly /cpu"
54+ TORCH_URL =" https://download.pytorch.org/whl/test /cpu"
6055
6156# pip packages needed by exir.
6257EXIR_REQUIREMENTS=(
63- torch==" 2.4.0. ${NIGHTLY_VERSION} "
64- torchvision==" 0.19.0. ${NIGHTLY_VERSION} " # For testing.
58+ torch==" 2.3.0 "
59+ torchvision==" 0.18.0 "
6560)
6661
6762# pip packages needed for development.
@@ -77,7 +72,7 @@ DEVEL_REQUIREMENTS=(
7772# TODO(dbort): Make each example publish its own requirements.txt
7873EXAMPLES_REQUIREMENTS=(
7974 timm==0.6.13
80- torchaudio==" 2.2.0. ${NIGHTLY_VERSION} "
75+ torchaudio==" 2.3.0 "
8176 torchsr==1.0.4
8277 transformers==4.38.2
8378)
@@ -92,7 +87,7 @@ REQUIREMENTS_TO_INSTALL=(
9287
9388# Install the requirements. `--extra-index-url` tells pip to look for package
9489# versions on the provided URL if they aren't available on the default URL.
95- pip install --extra-index-url " ${TORCH_NIGHTLY_URL } " \
90+ pip install --extra-index-url " ${TORCH_URL } " \
9691 " ${REQUIREMENTS_TO_INSTALL[@]} "
9792
9893#
0 commit comments