diff --git a/setup.py b/setup.py
index a1f62969..403a7dc8 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@
         "einops",
         "einops-exts",
         "transformers>=4.28.1",
-        "torch==2.0.1",
+        "torch==2.5.1",
         "pillow",
         "open_clip_torch>=2.16.0",
         "sentencepiece",