[code] update requirements

2023-05-06 21:05:53 +08:00 · 2023-05-06 21:05:53 +08:00 · 0324eb4103
parent 9b7f9b9d60
commit 0324eb4103
1 changed files with 2 additions and 2 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -1,5 +1,5 @@
 datasets==2.11.0
-deepspeed==0.9.1
-pytorch-lightning==2.0.2
+pytorch-lightning[strategies]==2.0.2
+tensorboardX==2.6
 torch==2.0.0
 transformers==4.28.1