[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-scm>=8
|
||||
torch==2.9.0
|
||||
torchvision
|
||||
torchaudio
|
||||
torchvision==0.24.0
|
||||
torchaudio==2.9.0
|
||||
wheel
|
||||
xgrammar>=0.1.30
|
||||
pandas-stubs
|
||||
|
||||
@@ -51,6 +51,9 @@ function install_binary_test() {
|
||||
pip install vllm=="${PIP_VLLM_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
|
||||
|
||||
|
||||
@@ -23,6 +23,7 @@ set -eo errexit
|
||||
|
||||
export VLLM_USE_MODELSCOPE=true
|
||||
export MODELSCOPE_HUB_FILE_LOCK=false
|
||||
export HF_HUB_OFFLINE=1
|
||||
|
||||
_info "====> Start Quickstart test"
|
||||
. "${SCRIPT_DIR}/doctests/001-quickstart-test.sh"
|
||||
|
||||
Reference in New Issue
Block a user