Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.

Commit 50ff602

Browse files
committed
Move backto nightly for torchtune
1 parent f7aef8a commit 50ff602

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

install/install_requirements.sh

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,10 @@ PYTORCH_NIGHTLY_VERSION=dev20240901
5252
# Nightly version for torchvision
5353
VISION_NIGHTLY_VERSION=dev20240901
5454

55+
# Nightly version for torchtune
56+
TUNE_NIGHTLY_VERSION=dev20240925
57+
58+
5559
# Uninstall triton, as nightly will depend on pytorch-triton, which is one and the same
5660
(
5761
set -x
@@ -72,6 +76,7 @@ fi
7276
REQUIREMENTS_TO_INSTALL=(
7377
torch=="2.5.0.${PYTORCH_NIGHTLY_VERSION}"
7478
torchvision=="0.20.0.${VISION_NIGHTLY_VERSION}"
79+
torchtune=="0.3.0.${TUNE_NIGHTLY_VERSION}"
7580
)
7681

7782
# Install the requirements. --extra-index-url tells pip to look for package

0 commit comments

Comments
 (0)