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

Update pyproject.toml for setup to work correctly

Browse Source
This commit is contained in:
powderluv
2022-05-01 21:02:22 -07:00
parent b8602d0b64
commit adb6b327dd
2 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pyproject.toml
View File

@@ -5,8 +5,8 @@ requires = [
"packaging",
"numpy",
"torch-mlir",
"iree-compiler",
"iree-runtime",
"torch-mlir>=20220428.420",
"iree-compiler>=20220427.13",
"iree-runtime>=20220427.13",
]
build-backend = "setuptools.build_meta"

7
setup_venv.sh
View File

@@ -24,7 +24,7 @@ source "$VENV_DIR/bin/activate" || die "Could not activate venv"
# Upgrade pip and install requirements. 'python' is used here in order to
# reference to the python executable from the venv.
python -m pip install --upgrade pip || die "Could not upgrade pip"
python -m pip install --upgrade -r "$TD/requirements.txt"
python -m pip install --upgrade -r "$TD/requirements.txt" --extra-index-url https://download.pytorch.org/whl/nightly/cpu -f https://github.com/llvm/torch-mlir/releases
python -m pip install --find-links https://github.com/llvm/torch-mlir/releases torch-mlir
python -m pip install --find-links https://github.com/NodLabs/SHARK/releases iree-compiler iree-runtime
#python -m pip install --find-links https://github.com/google/iree/releases iree-compiler iree-runtime
@@ -44,7 +44,10 @@ fi
python -m pip install transformers
python -m pip install git+https://github.com/pytorch/functorch.git
pip install . --extra-index-url https://download.pytorch.org/whl/nightly/cpu -f https://github.com/llvm/torch-mlir/releases
pip wheel -v -w $TD/wheelhouse $TD -f https://github.com/NodLabs/SHARK/releases -f https://github.com/llvm/torch-mlir/releases --extra-index-url https://download.pytorch.org/whl/nightly/cpu
python -m pip install . --extra-index-url https://download.pytorch.org/whl/nightly/cpu -f https://github.com/llvm/torch-mlir/releases -f https://github.com/NodLabs/SHARK/releases
echo "${Green}Before running examples activate venv with:"
echo " ${Green}source $VENV_DIR/bin/activate"