diff --git a/requirements.txt b/requirements.txt
index 0203294..ddc77c0 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,4 +3,5 @@ numpy
 torch>=2.2.0
 transformers>=4.43.0
 datasets
-einops>=0.8.0
\ No newline at end of file
+einops>=0.8.0
+vllm==0.6.2