[SHARK] Update dependencies

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
This commit is contained in:
Gaurav Shukla
2022-12-13 21:45:45 +05:30
parent 5f5e0766dd
commit b03038222d
4 changed files with 3 additions and 1 deletions

View File

@@ -28,6 +28,7 @@ Pillow
# web dependecies.
gradio
altair
# Testing and support.
#lit

View File

@@ -34,6 +34,7 @@ sacremoses
# web dependecies.
gradio
altair
scipy
#ONNX and ORT for benchmarking

View File

@@ -20,3 +20,4 @@ diffusers
scipy
ftfy
gradio
altair

View File

@@ -35,6 +35,5 @@ pip install --pre torch-mlir torch torchvision --extra-index-url https://downloa
pip install --upgrade -f https://nod-ai.github.io/SHARK-Runtime/pip-release-links.html iree-compiler iree-runtime
Write-Host "Building SHARK..."
pip install -e . -f https://llvm.github.io/torch-mlir/package-index/ -f https://nod-ai.github.io/SHARK-Runtime/pip-release-links.html
pip install diffusers transformers scipy pillow gradio
Write-Host "Build and installation completed successfully"
Write-Host "Source your venv with ./shark.venv/Scripts/activate"