colin
/
gpt-pretrain
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[code] update requirements

This commit is contained in:
Yiqing-Zhou 2023-05-06 21:05:53 +08:00
parent 9b7f9b9d60
commit 0324eb4103
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
requirements.txt
View File

@ -1,5 +1,5 @@
datasets==2.11.0 datasets==2.11.0
deepspeed==0.9.1 pytorch-lightning[strategies]==2.0.2
pytorch-lightning==2.0.2 tensorboardX==2.6
torch==2.0.0 torch==2.0.0
transformers==4.28.1 transformers==4.28.1