diff --git a/pyproject.toml b/pyproject.toml
index ebdc258..180b183 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,7 +16,7 @@ dependencies = [
     "matplotlib==3.9.0",
     "numpy==1.26.4",
     "positional-encodings==6.0.3",
-    "torch==2.4.0",
+    "torch==2.7.1",
     "tqdm==4.66.4",
     "vector-quantize-pytorch==1.14.24",
     "wandb==0.17.0",