Create requirements.txt
Browse files- requirements.txt +7 -0
requirements.txt
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
torch==2.2.1
|
| 2 |
+
transformers==4.38.2
|
| 3 |
+
datasets==2.18.0
|
| 4 |
+
evaluate==0.4.1
|
| 5 |
+
numpy==1.26.4
|
| 6 |
+
accelerate==0.28.0
|
| 7 |
+
bitsandbytes==0.43.0 # For 8-bit optimization (optional, reduces memory usage)
|