Tweak requirements.txt

requirements-importer-macos.txt

@@ -17,20 +17,20 @@ tqdm
 gin-config
 tensorflow-macos
 tensorflow-metal
-tf-models-nightly
+#tf-models-nightly
 #tensorflow-text-nightly
 transformers
-jax[cpu]
+#jax[cpu]
 
 # tflitehub dependencies.
 Pillow
 
 # Testing and support.
-lit
-pyyaml
+#lit
+#pyyaml
 
 #ONNX and ORT for benchmarking
---extra-index-url https://test.pypi.org/simple/
+#--extra-index-url https://test.pypi.org/simple/
 #protobuf
 #coloredlogs
 #flatbuffers

requirements-importer.txt

@@ -18,7 +18,7 @@ tensorflow
 tf-models-nightly
 tensorflow-text-nightly
 transformers
-jax[cpu]
+#jax[cpu]
 
 
 # tflitehub dependencies.

setup_venv.sh

@@ -95,11 +95,12 @@ echo "Installing ${RUNTIME}..."
 $PYTHON -m pip install --find-links https://github.com/${RUNTIME}/releases iree-compiler iree-runtime
 
 if [[ ! -z "${IMPORTER}" ]]; then
-  echo "${Yellow}Installing importer tools"
+  echo "${Yellow}Installing importer tools.."
   if [[ $(uname -s) = 'Linux' ]]; then
-    echo "${Yellow}Linux detected.. installing importer tools"
+    echo "${Yellow}Linux detected.. installing Linux 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/ --extra-index-url https://download.pytorch.org/whl/nightly/cpu
   elif [[ $(uname -s) = 'Darwin' ]]; then
+    echo "${Yellow}macOS detected.. installing macOS importer tools"
     #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 --extra-index-url https://download.pytorch.org/whl/nightly/cpu
   fi