Desident commited on
Commit
50ded61
·
verified ·
1 Parent(s): d475781

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -1,6 +1,7 @@
1
  pydantic==2.7.4
2
  fastapi==0.111.0
3
  --extra-index-url https://download.pytorch.org/whl/cu121
 
4
  torch==2.3.1+cu121
5
  torchaudio==2.3.1+cu121
6
  onnx==1.16.1
 
1
  pydantic==2.7.4
2
  fastapi==0.111.0
3
  --extra-index-url https://download.pytorch.org/whl/cu121
4
+ --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
5
  torch==2.3.1+cu121
6
  torchaudio==2.3.1+cu121
7
  onnx==1.16.1