mcps5601 commited on
Commit ·
5f83c67
1
Parent(s): d86a4e0
Add accelerate in requirements
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
|
@@ -3,4 +3,5 @@ transformers==4.36.2
|
|
| 3 |
torch==1.10.0+cu102
|
| 4 |
numpy==1.26.1
|
| 5 |
pandas==2.0.0
|
| 6 |
-
scikit-learn==1.2.2
|
|
|
|
|
|
| 3 |
torch==1.10.0+cu102
|
| 4 |
numpy==1.26.1
|
| 5 |
pandas==2.0.0
|
| 6 |
+
scikit-learn==1.2.2
|
| 7 |
+
accelerate==0.26.1
|