DeepLearning101 commited on
Commit
d3c88eb
·
verified ·
1 Parent(s): b77cc1b

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -3,7 +3,8 @@ roformer
3
  scikit-learn
4
  sentence-transformers
5
  sentencepiece
6
- torch==1.12.1
 
7
  transformers==4.21.2
8
  tqdm
9
  ujson
 
3
  scikit-learn
4
  sentence-transformers
5
  sentencepiece
6
+ torch==2.2.1
7
+ torchaudio==2.2.1
8
  transformers==4.21.2
9
  tqdm
10
  ujson