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

Commit 2ece601

Browse files
committed
Vision nightly is delayed
1 parent f58c22e commit 2ece601

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

install/install_requirements.sh

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ echo "Using pip executable: $PIP_EXECUTABLE"
6565
PYTORCH_NIGHTLY_VERSION=dev20241202
6666

6767
# Nightly version for torchvision
68-
VISION_NIGHTLY_VERSION=dev20241202
68+
VISION_NIGHTLY_VERSION=dev20241126
6969

7070
# Nightly version for torchtune
7171
TUNE_NIGHTLY_VERSION=dev20241126
@@ -92,7 +92,6 @@ fi
9292
# pip packages needed by exir.
9393
REQUIREMENTS_TO_INSTALL=(
9494
torch=="2.6.0.${PYTORCH_NIGHTLY_VERSION}"
95-
torchvision=="0.20.0.${VISION_NIGHTLY_VERSION}"
9695
torchtune=="0.5.0.${TUNE_NIGHTLY_VERSION}"
9796
)
9897

@@ -104,6 +103,13 @@ REQUIREMENTS_TO_INSTALL=(
104103
"${REQUIREMENTS_TO_INSTALL[@]}"
105104
)
106105

106+
# Workaround until torchvision nightly gets bumped
107+
(
108+
set -x
109+
$PIP_EXECUTABLE install --no-deps --extra-index-url "${TORCH_NIGHTLY_URL}" \
110+
torchvision=="0.20.0.${VISION_NIGHTLY_VERSION}"
111+
)
112+
107113
# For torchao need to install from github since nightly build doesn't have macos build.
108114
# TODO: Remove this and install nightly build, once it supports macos
109115
(

0 commit comments

Comments
 (0)