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

Fix macOS importer reqs

Browse Source
This commit is contained in:
Anush Elangovan
2022-06-12 01:45:05 -07:00
parent be66cc6ac1
commit 879942d87f
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
requirements-importer-macos.txt
View File

@@ -18,7 +18,7 @@ gin-config
tensorflow-macos tensorflow-macos
tensorflow-metal tensorflow-metal
tf-models-nightly tf-models-nightly
tensorflow-text-nightly #tensorflow-text-nightly
transformers transformers
jax[cpu] jax[cpu]

6
setup_venv.sh
View File

@@ -74,7 +74,7 @@ fi
$PYTHON -m pip install --upgrade pip || die "Could not upgrade pip" $PYTHON -m pip install --upgrade pip || die "Could not upgrade pip"
$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 --upgrade -r "$TD/requirements.txt" --extra-index-url https://download.pytorch.org/whl/nightly/cpu -f https://github.com/llvm/torch-mlir/releases
if [ "$torch_mlir_bin" = true ]; then if [ "$torch_mlir_bin" = true ]; then
$PYTHON -m pip install --find-links https://github.com/llvm/torch-mlir/releases torch-mlir $PYTHON -m pip install --find-links https://github.com/llvm/torch-mlir/releases torch-mlir --extra-index-url https://download.pytorch.org/whl/nightly/cpu
if [ $? -eq 0 ];then if [ $? -eq 0 ];then
echo "Successfully Installed torch-mlir" echo "Successfully Installed torch-mlir"
else else
@@ -98,10 +98,10 @@ if [[ ! -z "${IMPORTER}" ]]; then
echo "${Yellow}Installing importer tools" echo "${Yellow}Installing importer tools"
if [[ $(uname -s) = 'Linux' ]]; then if [[ $(uname -s) = 'Linux' ]]; then
echo "${Yellow}Linux detected.. installing importer tools" echo "${Yellow}Linux detected.. installing importer tools"
$PYTHON -m pip install --upgrade -r "$TD/requirements-importer.txt" -f https://github.com/${RUNTIME}/releases --extra-index-url https://test.pypi.org/simple/ $PYTHON -m pip install --upgrade -r "$TD/requirements-importer.txt" -f https://github.com/${RUNTIME}/releases --extra-index-url https://test.pypi.org/simple/ --extra-index-url https://download.pytorch.org/whl/nightly/cpu
elif [[ $(uname -s) = 'Darwin' ]]; then elif [[ $(uname -s) = 'Darwin' ]]; then
#Conda seems to have some problems installing these packages and hope they get resolved upstream. #Conda seems to have some problems installing these packages and hope they get resolved upstream.
$PYTHON -m pip install --upgrade -r "$TD/requirements-importer-macos.txt" -f https://github.com/${RUNTIME}/releases $PYTHON -m pip install --upgrade -r "$TD/requirements-importer-macos.txt" -f https://github.com/${RUNTIME}/releases --extra-index-url https://download.pytorch.org/whl/nightly/cpu
fi fi
fi fi