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