Skip to content

Commit

Permalink
Update install_requirements.sh
Browse files Browse the repository at this point in the history
  • Loading branch information
Jack-Khuu authored Nov 18, 2024
1 parent 8cb415d commit f9d0a29
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions install/install_requirements.sh
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@ echo "Using pip executable: $PIP_EXECUTABLE"
PYTORCH_NIGHTLY_VERSION=dev20241118

# Nightly version for torchvision
VISION_NIGHTLY_VERSION=dev20241118
VISION_NIGHTLY_VERSION=dev20241114

# Nightly version for torchtune
TUNE_NIGHTLY_VERSION=dev20241010
Expand All @@ -92,7 +92,6 @@ fi
# pip packages needed by exir.
REQUIREMENTS_TO_INSTALL=(
torch=="2.6.0.${PYTORCH_NIGHTLY_VERSION}"
torchvision=="0.20.0.${VISION_NIGHTLY_VERSION}"
torchtune=="0.4.0.${TUNE_NIGHTLY_VERSION}"
)

Expand All @@ -104,6 +103,13 @@ REQUIREMENTS_TO_INSTALL=(
"${REQUIREMENTS_TO_INSTALL[@]}"
)

# Workaround until torchvision nightly is gets bumped
(
set -x
$PIP_EXECUTABLE install --no-deps --extra-index-url "${TORCH_NIGHTLY_URL}" \
torchvision=="0.20.0.${VISION_NIGHTLY_VERSION}"
)

(
set -x
$PIP_EXECUTABLE install torchao=="0.5.0"
Expand Down

0 comments on commit f9d0a29

Please sign in to comment.