@@ -9,32 +9,32 @@ pip install --force-reinstall wheels/*.whl
9
9
10
10
if [[ ${GPU} == cuda* ]]; then
11
11
TARGET=cuda
12
- pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu123
12
+ pip install --pre -U --no-index - f https://mlc.ai/wheels mlc-ai-nightly-cu123
13
13
export LD_LIBRARY_PATH=/usr/local/cuda/compat/:$LD_LIBRARY_PATH
14
14
elif [[ ${GPU} == rocm* ]]; then
15
15
TARGET=rocm
16
- pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-rocm57
16
+ pip install --pre -U --no-index - f https://mlc.ai/wheels mlc-ai-nightly-rocm57
17
17
elif [[ ${GPU} == metal ]]; then
18
18
TARGET=metal
19
- pip install --pre -U --force-reinstal -f https://mlc.ai/wheels mlc-ai-nightly-cpu
19
+ pip install --pre -U --force-reinstall -f https://mlc.ai/wheels mlc-ai-nightly-cpu
20
20
elif [[ ${GPU} == wasm* ]]; then
21
21
TARGET=wasm
22
- pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cpu
22
+ pip install --pre -U --no-index - f https://mlc.ai/wheels mlc-ai-nightly-cpu
23
23
export TVM_SOURCE_DIR=$( dirname $( python -c ' import tvm; print(tvm.__file__)' ) )
24
24
export TVM_HOME=${TVM_SOURCE_DIR}
25
25
export MLC_LLM_SOURCE_DIR=$( pwd)
26
26
cd $TVM_SOURCE_DIR /web/ && make -j${NUM_THREADS} && cd -
27
27
cd $MLC_LLM_SOURCE_DIR /web/ && make -j${NUM_THREADS} && cd -
28
28
elif [[ ${GPU} == ios ]]; then
29
29
TARGET=ios
30
- pip install --pre -U --force-reinstal -f https://mlc.ai/wheels mlc-ai-nightly-cpu
30
+ pip install --pre -U --force-reinstall -f https://mlc.ai/wheels mlc-ai-nightly-cpu
31
31
elif [[ ${GPU} == android* ]]; then
32
32
TARGET=android
33
- pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cpu
33
+ pip install --pre -U --no-index - f https://mlc.ai/wheels mlc-ai-nightly-cpu
34
34
source /android_env_vars.sh
35
35
else
36
36
TARGET=vulkan
37
- pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cpu
37
+ pip install --pre -U --no-index - f https://mlc.ai/wheels mlc-ai-nightly-cpu
38
38
fi
39
39
40
40
python tests/python/integration/test_model_compile.py $TARGET $NUM_THREADS
0 commit comments