diff --git a/requirements.txt b/requirements.txt
index 94c9497..52ef9b8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,5 +2,9 @@ numpy
 Pillow
 Requests
 flash_attn
-transformers
-einops
\ No newline at end of file
+accelerate>=0.25.0
+huggingface-hub
+transformers>=4.44.0
+einops>=0.7.0
+torch>=2.1.2
+torchvision>=0.16.2
\ No newline at end of file