From 127ff8982eec0f5a9afe7c676608c1055aec7950 Mon Sep 17 00:00:00 2001 From: Yineng Zhang Date: Wed, 23 Apr 2025 02:17:21 -0700 Subject: [PATCH] fix torchvision::nms not exist (#5671) --- .github/workflows/pr-test-sgl-kernel.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/pr-test-sgl-kernel.yml b/.github/workflows/pr-test-sgl-kernel.yml index 0ab80bde6..9fd20bd9a 100644 --- a/.github/workflows/pr-test-sgl-kernel.yml +++ b/.github/workflows/pr-test-sgl-kernel.yml @@ -88,7 +88,7 @@ jobs: - name: Install run: | bash scripts/ci_install_dependency.sh - pip3 install torch==2.6.0 && pip3 install pytest + pip3 install torch==2.6.0 torchvision && pip3 install pytest pip3 uninstall sgl-kernel -y || true pip3 install sgl-kernel/dist/*whl --force-reinstall --no-deps pip3 list | grep sgl-kernel