Update prep_deps to include cargo installation script (octoml#220)

Lunderberg · May 23, 2023 · 1f53191 · 1f53191
1 parent 110c6d3
commit 1f53191
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 23 deletions.
diff --git a/scripts/gh_deploy_site.sh b/scripts/gh_deploy_site.sh
diff --git a/scripts/local_deploy_site.sh b/scripts/local_deploy_site.sh
diff --git a/scripts/prep_deps.sh b/scripts/prep_deps.sh
@@ -1,9 +1,22 @@
 #!/bin/bash
 # This file prepares all the necessary dependencies for the web build.
-set -euxo pipefail
+set -uxo pipefail
 
 # need rust for tokenizers
-cargo --version
+cargo version
+CARGO_RESULT=$?
+if [ $CARGO_RESULT -eq 0 ]; then
+  echo "Cargo installed"
+else
+  printf "Cargo is required to compile tokenizers in MLC-LLM, do you want to install cargo (y/n)?"
+  read answer
+  if [ "$answer" != "${answer#[Yy]}" ] ;then 
+    curl https://sh.rustup.rs -sSf | sh
+  else
+    echo "Exit installation."
+    exit 1
+  fi
+fi
 
 TVM_HOME_SET="${TVM_HOME:-}"