Merge pull request meta-llama#207 from hyungupark/patch-1

Fix issue CPU_ARCH not found in download.sh
Dholvangha · May 14, 2024 · f37d7d3 · f37d7d3
2 parents cc44ca2 + d587ae0
commit f37d7d3
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)