Add OSX-importer requirements

requirements-importer-macos.txt

@@ -0,0 +1,38 @@
+-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
+--pre
+
+numpy
+torch
+torchvision
+
+#iree-compiler  | iree-runtime should already be installed
+#these dont work ok osx
+#iree-tools-tflite
+#iree-tools-xla
+#iree-tools-tf
+
+# TensorFlow and JAX.
+gin-config
+tensorflow-macos
+tensorflow-metal
+tf-models-nightly
+tensorflow-text-nightly
+transformers
+jax[cpu]
+
+# tflitehub dependencies.
+Pillow
+
+# Testing and support.
+lit
+pyyaml
+
+#ONNX and ORT for benchmarking
+--extra-index-url https://test.pypi.org/simple/
+#protobuf
+#coloredlogs
+#flatbuffers
+#sympy
+#psutil
+#onnx-weekly
+#ort-nightly

requirements-importer.txt

@@ -1,5 +1,11 @@
-#iree-compiler should already be installed
-#iree-runtime
+-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
+--pre
+
+numpy
+torch
+torchvision
+
+#iree-compiler  | iree-runtime should already be installed
 iree-tools-tflite
 iree-tools-xla
 iree-tools-tf

requirements.txt

@@ -1,11 +1,3 @@
--f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
---pre
-
-numpy
-torch
-torchvision
-
-# Build requirements.
 setuptools
 wheel
 pytest

setup_venv.sh

@@ -100,8 +100,8 @@ if [[ ! -z "${IMPORTER}" ]]; then
     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/
   elif [[ $(uname -s) = 'Darwin' ]]; then
-    #Conda seems to have some problems install these packages and hope they get resolved upstream.
-    $PYTHON -m pip install tensorflow-macos tensorflow-metal
+    #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
   fi
 fi