Skip to content

Commit 3341243

Browse files
committed
Install torchao from gh
1 parent 1559386 commit 3341243

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

install_requirements.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,10 @@ REQUIREMENTS_TO_INSTALL=(
7171
# versions on the provided URL if they aren't available on the default URL.
7272
$PIP_EXECUTABLE install --extra-index-url "${TORCH_NIGHTLY_URL}" \
7373
"${REQUIREMENTS_TO_INSTALL[@]}"
74+
75+
# For torchao need to install from github since nightly build doesn't have macos build.
76+
# TODO: Remove this and install nightly build, once it supports macos
77+
$PIP_EXECUTABLE install git+https://github.com/pytorch/[email protected]
7478
if [[ -x "$(command -v nvidia-smi)" ]]; then
7579
$PYTHON_EXECUTABLE scripts/patch_triton.py
7680
fi

0 commit comments

Comments
 (0)