Skip to content

Commit fd1857a

Browse files
authored
Update install_requirements.sh to tune + pt/pt dev20241218 (#1426)
* Update install_requirements.sh to pt/pt dev20241217 * Update install_requirements.sh to 1218 * Update install_requirements.sh * Rearrange install order; previuosly inconsistent extra-index-url
1 parent 113e40b commit fd1857a

File tree

1 file changed

+13
-14
lines changed

1 file changed

+13
-14
lines changed

install/install_requirements.sh

+13-14
Original file line numberDiff line numberDiff line change
@@ -44,31 +44,20 @@ fi
4444

4545
echo "Using pip executable: $PIP_EXECUTABLE"
4646

47-
#
48-
# First install requirements in install/requirements.txt. Older torch may be
49-
# installed from the dependency of other models. It will be overridden by
50-
# newer version of torch nightly installed later in this script.
51-
#
52-
53-
(
54-
set -x
55-
$PIP_EXECUTABLE install -r install/requirements.txt --extra-index-url https://download.pytorch.org/whl/nightly/cu121
56-
)
57-
5847
# Since torchchat often uses main-branch features of pytorch, only the nightly
5948
# pip versions will have the required features. The PYTORCH_NIGHTLY_VERSION value should
6049
# agree with the third-party/pytorch pinned submodule commit.
6150
#
6251
# NOTE: If a newly-fetched version of the executorch repo changes the value of
6352
# PYTORCH_NIGHTLY_VERSION, you should re-run this script to install the necessary
6453
# package versions.
65-
PYTORCH_NIGHTLY_VERSION=dev20241213
54+
PYTORCH_NIGHTLY_VERSION=dev20241218
6655

6756
# Nightly version for torchvision
68-
VISION_NIGHTLY_VERSION=dev20241213
57+
VISION_NIGHTLY_VERSION=dev20241218
6958

7059
# Nightly version for torchtune
71-
TUNE_NIGHTLY_VERSION=dev20241126
60+
TUNE_NIGHTLY_VERSION=dev20241218
7261

7362
# Uninstall triton, as nightly will depend on pytorch-triton, which is one and the same
7463
(
@@ -96,6 +85,16 @@ REQUIREMENTS_TO_INSTALL=(
9685
torchtune=="0.5.0.${TUNE_NIGHTLY_VERSION}"
9786
)
9887

88+
#
89+
# First install requirements in install/requirements.txt. Older torch may be
90+
# installed from the dependency of other models. It will be overridden by
91+
# newer version of torch nightly installed later in this script.
92+
#
93+
(
94+
set -x
95+
$PIP_EXECUTABLE install -r install/requirements.txt --extra-index-url "${TORCH_NIGHTLY_URL}"
96+
)
97+
9998
# Install the requirements. --extra-index-url tells pip to look for package
10099
# versions on the provided URL if they aren't available on the default URL.
101100
(

0 commit comments

Comments
 (0)