diff --git a/models/llama/requirements.txt b/models/llama/requirements.txt
index c4902e2..0cb7b98 100644
--- a/models/llama/requirements.txt
+++ b/models/llama/requirements.txt
@@ -4,4 +4,4 @@ numpy
 peft
 transformers==4.35.1
 sentencepiece==0.1.99
-torch==1.13.1
\ No newline at end of file
+torch
diff --git a/models/long_roberta/requirements.txt b/models/long_roberta/requirements.txt
index 92de230..9f083f4 100644
--- a/models/long_roberta/requirements.txt
+++ b/models/long_roberta/requirements.txt
@@ -7,7 +7,7 @@ pandas
 datasets
 protobuf
 matplotlib
-torch==1.13.1
+torch
 accelerate
 pyarrow
-pytest
\ No newline at end of file
+pytest