Update download.sh

modify for CPU_ARCH not found
AlexYangLe · May 14, 2024 · d587ae0 · d587ae0
1 parent cc44ca2
commit d587ae0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/download.sh b/download.sh
@@ -51,7 +51,8 @@ do
  wget --continue ${PRESIGNED_URL/'*'/"${MODEL_PATH}/tokenizer.model"} -O ${TARGET_FOLDER}"/${MODEL_FOLDER_PATH}/tokenizer.model"
  wget --continue ${PRESIGNED_URL/'*'/"${MODEL_PATH}/checklist.chk"} -O ${TARGET_FOLDER}"/${MODEL_FOLDER_PATH}/checklist.chk"
  echo "Checking checksums"
- if [ "$CPU_ARCH" = "arm64" ]; then
+ CPU_ARCH=$(uname -m)
+ if [[ "$CPU_ARCH" == "arm64" ]]; then
  (cd ${TARGET_FOLDER}"/${MODEL_FOLDER_PATH}" && md5 checklist.chk)
  else
  (cd ${TARGET_FOLDER}"/${MODEL_FOLDER_PATH}" && md5sum -c checklist.chk)