zhu-mingye commited on
Commit ·
490d094
1
Parent(s): c2293c5
Upgrade transformers and torch for CodeT5+ support
Browse files- requirements.txt +4 -5
requirements.txt
CHANGED
|
@@ -1,6 +1,5 @@
|
|
| 1 |
-
transformers=
|
| 2 |
-
torch=
|
| 3 |
sentencepiece>=0.1.96
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
datasets>=2.0.0
|
|
|
|
| 1 |
+
transformers>=4.40.0
|
| 2 |
+
torch>=2.0.0
|
| 3 |
sentencepiece>=0.1.96
|
| 4 |
+
accelerate>=0.20.0
|
| 5 |
+
datasets>=2.0.0
|
|
|