[cherry-pick][CI] Enforce torchaudio and torchvison compatible with pta (#7688)
### What this PR does / why we need it? This patch cherry-pick from #7648 Signed-off-by: wangli <wangli858794774@gmail.com>
This commit is contained in:
@@ -14,8 +14,8 @@ psutil
|
|||||||
setuptools>=64
|
setuptools>=64
|
||||||
setuptools-scm>=8
|
setuptools-scm>=8
|
||||||
torch==2.9.0
|
torch==2.9.0
|
||||||
torchvision
|
torchvision==0.24.0
|
||||||
torchaudio
|
torchaudio==2.9.0
|
||||||
wheel
|
wheel
|
||||||
xgrammar>=0.1.30
|
xgrammar>=0.1.30
|
||||||
pandas-stubs
|
pandas-stubs
|
||||||
|
|||||||
@@ -51,6 +51,9 @@ function install_binary_test() {
|
|||||||
pip install vllm=="${PIP_VLLM_VERSION}"
|
pip install vllm=="${PIP_VLLM_VERSION}"
|
||||||
|
|
||||||
pip install vllm-ascend=="${PIP_VLLM_ASCEND_VERSION}"
|
pip install vllm-ascend=="${PIP_VLLM_ASCEND_VERSION}"
|
||||||
|
if [ "${PIP_VLLM_ASCEND_VERSION}" == "0.17.0rc1" ]; then
|
||||||
|
pip install torchvision==0.24.0 torchaudio==2.9.0
|
||||||
|
fi
|
||||||
|
|
||||||
pip list | grep vllm
|
pip list | grep vllm
|
||||||
|
|
||||||
|
|||||||
@@ -23,6 +23,7 @@ set -eo errexit
|
|||||||
|
|
||||||
export VLLM_USE_MODELSCOPE=true
|
export VLLM_USE_MODELSCOPE=true
|
||||||
export MODELSCOPE_HUB_FILE_LOCK=false
|
export MODELSCOPE_HUB_FILE_LOCK=false
|
||||||
|
export HF_HUB_OFFLINE=1
|
||||||
|
|
||||||
_info "====> Start Quickstart test"
|
_info "====> Start Quickstart test"
|
||||||
. "${SCRIPT_DIR}/doctests/001-quickstart-test.sh"
|
. "${SCRIPT_DIR}/doctests/001-quickstart-test.sh"
|
||||||
|
|||||||
Reference in New Issue
Block a user