49
49
# Since ExecuTorch often uses main-branch features of pytorch, only the nightly
50
50
# pip versions will have the required features. The NIGHTLY_VERSION value should
51
51
# 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
57
52
58
53
# 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"
60
55
61
56
# pip packages needed by exir.
62
57
EXIR_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 "
65
60
)
66
61
67
62
# pip packages needed for development.
@@ -77,7 +72,7 @@ DEVEL_REQUIREMENTS=(
77
72
# TODO(dbort): Make each example publish its own requirements.txt
78
73
EXAMPLES_REQUIREMENTS=(
79
74
timm==0.6.13
80
- torchaudio==" 2.2.0. ${NIGHTLY_VERSION} "
75
+ torchaudio==" 2.3.0 "
81
76
torchsr==1.0.4
82
77
transformers==4.38.2
83
78
)
@@ -92,7 +87,7 @@ REQUIREMENTS_TO_INSTALL=(
92
87
93
88
# Install the requirements. `--extra-index-url` tells pip to look for package
94
89
# 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 } " \
96
91
" ${REQUIREMENTS_TO_INSTALL[@]} "
97
92
98
93
#
0 commit comments