github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-08 05:24:00 -05:00
Code Issues Packages Projects Releases Wiki Activity

Update setup_venv.sh

Browse Source
This commit is contained in:
Ean Garvey
2024-05-28 18:25:54 -05:00
committed by GitHub
parent 78b6e4ff4b
commit 25e4c4077e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup_venv.sh
View File

@@ -54,7 +54,7 @@ PYTORCH_URL="https://download.pytorch.org/whl/nightly/cpu/"
# Upgrade pip and install requirements.
$PYTHON -m pip install --upgrade pip || die "Could not upgrade pip"
$PYTHON -m pip install --upgrade --pre torch torchvision torchaudio --index-url $PYTORCH_URL
$PYTHON -m pip install --upgrade --pre torch==2.3.0 torchvision torchaudio --index-url $PYTORCH_URL
if [[ -z "${NO_BREVITAS}" ]]; then
$PYTHON -m pip install git+https://github.com/Xilinx/brevitas.git@dev
fi