Commit
•
66af894
1
Parent(s):
b0b2ca0
Update requirements.txt
Browse files- requirements.txt +3 -0
requirements.txt
CHANGED
@@ -2,6 +2,9 @@ transformers>=4.35.0
|
|
2 |
numba
|
3 |
numpy
|
4 |
torch
|
|
|
|
|
|
|
5 |
tqdm
|
6 |
more-itertools
|
7 |
tiktoken
|
|
|
2 |
numba
|
3 |
numpy
|
4 |
torch
|
5 |
+
torch>=1.13.1,<=2.0.1
|
6 |
+
torchvision>=0.14.1,<=0.15.2
|
7 |
+
torchaudio>=0.13.1,<=2.0.2
|
8 |
tqdm
|
9 |
more-itertools
|
10 |
tiktoken
|