From 6c97bd9bc1b4135c952162ecb834b850ec3b1428 Mon Sep 17 00:00:00 2001 From: vmpuri Date: Tue, 15 Oct 2024 02:40:29 -0700 Subject: [PATCH] Update torchtune pin to 0.4.0-dev20241010 --- install/install_requirements.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/install/install_requirements.sh b/install/install_requirements.sh index a05e255db..6344509d8 100755 --- a/install/install_requirements.sh +++ b/install/install_requirements.sh @@ -53,7 +53,7 @@ PYTORCH_NIGHTLY_VERSION=dev20241002 VISION_NIGHTLY_VERSION=dev20241002 # Nightly version for torchtune -TUNE_NIGHTLY_VERSION=dev20240928 +TUNE_NIGHTLY_VERSION=dev20241010 # Uninstall triton, as nightly will depend on pytorch-triton, which is one and the same ( @@ -78,7 +78,7 @@ fi REQUIREMENTS_TO_INSTALL=( torch=="2.6.0.${PYTORCH_NIGHTLY_VERSION}" torchvision=="0.20.0.${VISION_NIGHTLY_VERSION}" - torchtune=="0.3.0.${TUNE_NIGHTLY_VERSION}" + torchtune=="0.4.0.${TUNE_NIGHTLY_VERSION}" ) # Install the requirements. --extra-index-url tells pip to look for package