diff --git a/install/install_requirements.sh b/install/install_requirements.sh index bb0190a29..b75674edc 100755 --- a/install/install_requirements.sh +++ b/install/install_requirements.sh @@ -52,6 +52,10 @@ PYTORCH_NIGHTLY_VERSION=dev20240901 # Nightly version for torchvision VISION_NIGHTLY_VERSION=dev20240901 +# Nightly version for torchtune +TUNE_NIGHTLY_VERSION=dev20240925 + + # Uninstall triton, as nightly will depend on pytorch-triton, which is one and the same ( set -x @@ -72,6 +76,7 @@ fi REQUIREMENTS_TO_INSTALL=( torch=="2.5.0.${PYTORCH_NIGHTLY_VERSION}" torchvision=="0.20.0.${VISION_NIGHTLY_VERSION}" + torchtune=="0.3.0.${TUNE_NIGHTLY_VERSION}" ) # Install the requirements. --extra-index-url tells pip to look for package