Skip to content

Commit 127ff89

Browse files
authored
fix torchvision::nms not exist (#5671)
1 parent 8777a1d commit 127ff89

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ jobs:
8888
- name: Install
8989
run: |
9090
bash scripts/ci_install_dependency.sh
91-
pip3 install torch==2.6.0 && pip3 install pytest
91+
pip3 install torch==2.6.0 torchvision && pip3 install pytest
9292
pip3 uninstall sgl-kernel -y || true
9393
pip3 install sgl-kernel/dist/*whl --force-reinstall --no-deps
9494
pip3 list | grep sgl-kernel

0 commit comments

Comments
 (0)