From 5d13544fa28da2fd1d9ff26f76de350745c717c9 Mon Sep 17 00:00:00 2001
From: JJGadgets <git@jjgadgets.tech>
Date: Sun, 13 Oct 2024 03:57:35 +0800
Subject: [PATCH] chore: cleanup

---
 apps/mlc-llm/ci/latest.sh | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/apps/mlc-llm/ci/latest.sh b/apps/mlc-llm/ci/latest.sh
index ea3f2d4e..e78d1180 100755
--- 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