fix bugs

Author: zewenli98

dev_dep_versions.yml

@@ -1,4 +1,4 @@
-__version__: "2.3.0.dev0"
+__version__: "2.4.0.dev0"
 __cuda_version__: "12.1"
 __cudnn_version__: "8.9"
 __tensorrt_version__: "10.0.1.6"

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