diff --git a/requirements.txt b/requirements.txt
index 9ac8f4f..f3c7280 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,7 +11,7 @@ pytorch-lightning==1.8.5
 pyyaml>=6.0.1
 tokenizers>=0.19,<0.20
 --extra-index-url https://download.pytorch.org/whl/cu117
-torch==1.13.1+cu117
+torch==2.6.0
 torchvision==0.14.1+cu117
 tqdm>=4.65.0
 transformers @ git+https://github.com/huggingface/transformers.git@v4.42.4