Skip to content

Commit 1d0f267

Browse files
authored
Update install_requirements.sh
1 parent dc0c550 commit 1d0f267

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

install/install_requirements.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,13 +81,13 @@ then
8181
REQUIREMENTS_TO_INSTALL=(
8282
torch=="2.7.0.${PYTORCH_NIGHTLY_VERSION}"
8383
torchvision=="0.22.0.${VISION_NIGHTLY_VERSION}"
84-
torchtune=="0.5.0"
84+
torchtune=="0.6.0"
8585
)
8686
else
8787
REQUIREMENTS_TO_INSTALL=(
8888
torch=="2.7.0.${PYTORCH_NIGHTLY_VERSION}"
8989
torchvision=="0.22.0.${VISION_NIGHTLY_VERSION}"
90-
torchtune=="0.5.0.${TUNE_NIGHTLY_VERSION}"
90+
torchtune=="0.6.0.${TUNE_NIGHTLY_VERSION}"
9191
)
9292
fi
9393

0 commit comments

Comments
 (0)