diff --git a/pyproject.toml b/pyproject.toml
index a5f3c85..567b59a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,7 +13,7 @@ classifiers = [
     "License :: OSI Approved :: Apache Software License",
 ]
 dependencies = [
-    "torch==2.3.1",
+    "torch==2.6.0",
     "torchvision==0.18.1", 
     "pytorch-lightning>=1.0.8",
     "transformers>=4.49.0",