diff --git a/requirements.txt b/requirements.txt
index 08a6d2c..d6d2532 100644
--- 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