gz412 commited on
Commit
3c2217a
·
1 Parent(s): 1105c2a

fix requirments

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -1,3 +1,5 @@
 
 
1
  conformer==0.3.2
2
  deepspeed==0.15.1; sys_platform == 'linux'
3
  diffusers==0.29.0
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu121
2
+ --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ # https://github.com/microsoft/onnxruntime/issues/21684
3
  conformer==0.3.2
4
  deepspeed==0.15.1; sys_platform == 'linux'
5
  diffusers==0.29.0