EngineX/xc-llm-ascend
3
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Set torchvision<0.21.0 to match torch/torch_npu version (#479)

Browse Source 
### 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>
This commit is contained in:
Yikun Jiang
2025-04-08 09:15:42 +08:00
committed by GitHub
parent 1d88dacf9f
commit 579d858a20
2 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pyproject.toml
View File

@@ -11,5 +11,6 @@ requires = [
"setuptools-scm>=8", "setuptools-scm>=8",
"torch_npu >= 2.5.1rc1", "torch_npu >= 2.5.1rc1",
"torch >= 2.5.1", "torch >= 2.5.1",
"torchvision<0.21.0",
] ]
build-backend = "setuptools.build_meta" build-backend = "setuptools.build_meta"

1
requirements.txt
View File

@@ -9,3 +9,4 @@ setuptools>=64
setuptools-scm>=8 setuptools-scm>=8
torch_npu >= 2.5.1rc1 torch_npu >= 2.5.1rc1
torch >= 2.5.1 torch >= 2.5.1
torchvision<0.21.0