fix torchvision::nms not exist (#5671)

127ff898 · Yineng Zhang · GitHub · 8777a1d2 · 127ff898
Unverified Commit 127ff898 authored Apr 23, 2025 by Yineng Zhang Committed by GitHub Apr 23, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

.github/workflows/pr-test-sgl-kernel.yml .github/workflows/pr-test-sgl-kernel.yml +1 -1

No files found.
--- 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