zhu-mingye commited on
Commit
490d094
·
1 Parent(s): c2293c5

Upgrade transformers and torch for CodeT5+ support

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -5
requirements.txt CHANGED
@@ -1,6 +1,5 @@
1
- transformers==4.21.3
2
- torch==1.13.1
3
  sentencepiece>=0.1.96
4
- tokenizers>=0.12.1,<0.14
5
- accelerate>=0.12.0,<0.21
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