AnhP commited on
Commit
ef3cc56
·
verified ·
1 Parent(s): a123cbb

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -5
requirements.txt CHANGED
@@ -1,7 +1,9 @@
1
  # Platform-specific requirements section
2
- pip>=23.3,<24.1
3
- onnxruntime; sys_platform == 'darwin'
4
- onnxruntime-gpu; sys_platform != 'darwin'
 
 
5
 
6
  # Core dependencies
7
  PyYAML
@@ -12,8 +14,7 @@ torchaudio==2.3.1
12
  faiss-cpu>=1.7.3
13
 
14
  # Machine learning, NLP and deep learning
15
- fairseq==0.12.2; python_version == '3.10'
16
- fairseq @ git+https://github.com/One-sixth/fairseq.git; python_version > '3.10'
17
  scikit-learn
18
  einops
19
 
 
1
  # Platform-specific requirements section
2
+ pip==23.3
3
+ wheel
4
+ ort-nightly-gpu --index-url=https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ort-cuda-12-nightly/pypi/simple/
5
+ onnxruntime==1.20.0; sys_platform == 'darwin'
6
+ onnxruntime-gpu==1.20.0; sys_platform != 'darwin'
7
 
8
  # Core dependencies
9
  PyYAML
 
14
  faiss-cpu>=1.7.3
15
 
16
  # Machine learning, NLP and deep learning
17
+ fairseq @ git+https://github.com/One-sixth/fairseq.git
 
18
  scikit-learn
19
  einops
20