From 579d858a2075809b55aeb10e8f6188bc88708831 Mon Sep 17 00:00:00 2001
From: Yikun Jiang <yikunkero@gmail.com>
Date: Tue, 8 Apr 2025 09:15:42 +0800
Subject: [PATCH] Set torchvision<0.21.0 to match torch/torch_npu version
 (#479)

### 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>
---
 pyproject.toml   | 1 +
 requirements.txt | 1 +
 2 files changed, 2 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index 35d7ea5..c73b9b3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -11,5 +11,6 @@ requires = [
     "setuptools-scm>=8",
     "torch_npu >= 2.5.1rc1",
     "torch >= 2.5.1",
+    "torchvision<0.21.0",
 ]
 build-backend = "setuptools.build_meta"
diff --git a/requirements.txt b/requirements.txt
index 1ed8172..3f3c0b0 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,3 +9,4 @@ setuptools>=64
 setuptools-scm>=8
 torch_npu >= 2.5.1rc1
 torch >= 2.5.1
+torchvision<0.21.0