579d858a20
### What this PR does / why we need it? Set torchvision<0.21.0 to match torch/torch_npu version to resolve `RuntimeError: operator torchvision::nms does not exist`. Closes: https://github.com/vllm-project/vllm-ascend/issues/477 ### Does this PR introduce _any_ user-facing change? No ### How was this patch tested? CI passed Signed-off-by: Yikun Jiang <yikunkero@gmail.com>
17 lines
333 B
TOML
17 lines
333 B
TOML
[build-system]
|
|
# Should be mirrored in requirements.txt
|
|
requires = [
|
|
"cmake>=3.26",
|
|
"decorator",
|
|
"numpy<2.0.0",
|
|
"pybind11",
|
|
"pyyaml",
|
|
"scipy",
|
|
"setuptools>=64",
|
|
"setuptools-scm>=8",
|
|
"torch_npu >= 2.5.1rc1",
|
|
"torch >= 2.5.1",
|
|
"torchvision<0.21.0",
|
|
]
|
|
build-backend = "setuptools.build_meta"
|