Wuvin commited on
Commit
b898902
1 Parent(s): 5cda7f6
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  pytorch3d @ https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt221/pytorch3d-0.7.6-cp310-cp310-linux_x86_64.whl
2
- ort_nightly_gpu==1.17.0.dev20240118002 --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ort-cuda-12-nightly/pypi/simple/
3
- onnxruntime_gpu==1.17.0 --index-url https://pkgs.dev.azure.com/onnxruntime/onnxruntime/_packaging/onnxruntime-cuda-12/pypi/simple/
4
  accelerate
5
  datasets
6
  diffusers>=0.26.3
 
1
  pytorch3d @ https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt221/pytorch3d-0.7.6-cp310-cp310-linux_x86_64.whl
2
+ ort_nightly_gpu # ==1.17.0.dev20240118002 --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ort-cuda-12-nightly/pypi/simple/
3
+ onnxruntime_gpu==1.17.0 # --index-url https://pkgs.dev.azure.com/onnxruntime/onnxruntime/_packaging/onnxruntime-cuda-12/pypi/simple/
4
  accelerate
5
  datasets
6
  diffusers>=0.26.3