fix bugs

zewenli98 · zewenli98 · commit bc9d93dfd4ca · 2024-04-27T16:21:52.000-07:00
diff --git a/py/requirements.txt b/py/requirements.txt
@@ -1,9 +1,9 @@
 numpy
 packaging
 pybind11==2.6.2
---extra-index-url https://download.pytorch.org/whl/cu121
-torch==2.3.0
-torchvision==0.18.0
+--extra-index-url https://download.pytorch.org/whl/nightly/cu121
+torch>=2.4.0.dev,<2.5.0
+torchvision>=0.19.0.dev,<0.20.0
 --extra-index-url https://pypi.ngc.nvidia.com
 pyyaml
 tensorrt==10.0.1.6
