diff --git a/finetune/requirements.txt b/finetune/requirements.txt
index bc1d43e..eefb87b 100644
--- a/finetune/requirements.txt
+++ b/finetune/requirements.txt
@@ -5,6 +5,6 @@ accelerate==0.24.1
 attrdict
 tqdm
 
-deepspeed==0.12.2
-datasets
+deepspeed>=0.19.0
+datasets>=4.8.5
 tensorboardX
diff --git a/requirements.txt b/requirements.txt
index 8ac4fff..c1a0f64 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,8 +1,12 @@
-torch>=2.0
-tokenizers>=0.14.0
-transformers==4.35.0
-accelerate
-sympy==1.12
+torch>=2.12.0
+torchvision>=0.27.0
+torchaudio>=2.11.0
+numpy>=2.0
+tokenizers>=0.22.0
+transformers>=5.8.1
+accelerate>=1.13.0
+safetensors>=0.7.0
+sympy>=1.12
 pebble 
 timeout-decorator 
 attrdict