Skip to content

Commit

Permalink
Update install_requirements.sh to tune + pt/pt dev20241218 (#1426)
Browse files Browse the repository at this point in the history
* 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
  • Loading branch information
Jack-Khuu authored Dec 19, 2024
1 parent 113e40b commit fd1857a
Showing 1 changed file with 13 additions and 14 deletions.
27 changes: 13 additions & 14 deletions install/install_requirements.sh
Original file line number Diff line number Diff line change
Expand Up @@ -44,31 +44,20 @@ fi

echo "Using pip executable: $PIP_EXECUTABLE"

#
# First install requirements in install/requirements.txt. Older torch may be
# installed from the dependency of other models. It will be overridden by
# newer version of torch nightly installed later in this script.
#

(
set -x
$PIP_EXECUTABLE install -r install/requirements.txt --extra-index-url https://download.pytorch.org/whl/nightly/cu121
)

# Since torchchat often uses main-branch features of pytorch, only the nightly
# pip versions will have the required features. The PYTORCH_NIGHTLY_VERSION value should
# agree with the third-party/pytorch pinned submodule commit.
#
# NOTE: If a newly-fetched version of the executorch repo changes the value of
# PYTORCH_NIGHTLY_VERSION, you should re-run this script to install the necessary
# package versions.
PYTORCH_NIGHTLY_VERSION=dev20241213
PYTORCH_NIGHTLY_VERSION=dev20241218

# Nightly version for torchvision
VISION_NIGHTLY_VERSION=dev20241213
VISION_NIGHTLY_VERSION=dev20241218

# Nightly version for torchtune
TUNE_NIGHTLY_VERSION=dev20241126
TUNE_NIGHTLY_VERSION=dev20241218

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

#
# First install requirements in install/requirements.txt. Older torch may be
# installed from the dependency of other models. It will be overridden by
# newer version of torch nightly installed later in this script.
#
(
set -x
$PIP_EXECUTABLE install -r install/requirements.txt --extra-index-url "${TORCH_NIGHTLY_URL}"
)

# Install the requirements. --extra-index-url tells pip to look for package
# versions on the provided URL if they aren't available on the default URL.
(
Expand Down

0 comments on commit fd1857a

Please sign in to comment.