chore: cleanup

JJGadgets · Oct 12, 2024 · 5d13544 · 5d13544
1 parent ca5f398
commit 5d13544
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/apps/mlc-llm/ci/latest.sh b/apps/mlc-llm/ci/latest.sh
@@ -5,8 +5,12 @@ channel=$1
 
 if [[ "${channel}" == "stable" ]]; then
     version="$(curl -s https://mlc.ai/wheels | grep -oE '>mlc_llm_cpu-(.*)-cp311-manylinux.*_x86_64.whl' | awk -F'-' '{print $2}' | tail -n 1)"
+elif [[ "${channel}" == "cuda" ]]; then
+    version="$(curl -s https://mlc.ai/wheels | grep -oE '>mlc_llm_cpu-(.*)-cp311-manylinux.*_x86_64.whl' | awk -F'-' '{print $2}' | tail -n 1)"
 elif [[ "${channel}" == "nightly" ]]; then
     version="$(date +%Y.%m.%d)"
+elif [[ "${channel}" == "cuda-nightly" ]]; then
+    version="$(date +%Y.%m.%d)"
 else
     echo 'No version could be extracted!'
     exit 1