diff --git a/prompt_compression/requirements.txt b/prompt_compression/requirements.txt
index 7d87363..3b1a408 100644
--- a/prompt_compression/requirements.txt
+++ b/prompt_compression/requirements.txt
@@ -11,7 +11,7 @@ rouge==1.0.1
 sentencepiece==0.2.0
 tiktoken==0.8.0
 tokenizers==0.19.1
-torch==2.1.1
+torch==2.8.0
 tqdm==4.67.0
 transformers==4.43.1
 wandb==0.18.6