Add requirements
Browse files- requirements.txt +9 -0
requirements.txt
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
torch>=2.0.0
|
| 2 |
+
transformers>=4.30.0
|
| 3 |
+
datasets>=2.14.0
|
| 4 |
+
huggingface_hub>=0.16.0
|
| 5 |
+
einops>=0.6.1
|
| 6 |
+
tqdm>=4.65.0
|
| 7 |
+
numpy>=1.24.0
|
| 8 |
+
matplotlib>=3.7.0
|
| 9 |
+
scipy>=1.10.0
|