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
|