diff --git a/Dockerfile b/Dockerfile
index ab1f969..4119f2e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -18,7 +18,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
     if [ "$TARGETPLATFORM" = "linux/amd64" ]; then \
       python3 -m pip install \
         --no-cache-dir --disable-pip-version-check --extra-index-url https://download.pytorch.org/whl/cpu  \
-        torch==2.3.1+cpu .[all]; \
+        torch==2.5.1+cpu .[all]; \
     else \
       python3 -m pip install \
         --no-cache-dir --disable-pip-version-check \
diff --git a/Dockerfile.wolfi b/Dockerfile.wolfi
index 3d94ffd..d54be8f 100644
--- a/Dockerfile.wolfi
+++ b/Dockerfile.wolfi
@@ -13,7 +13,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
     if [ "$TARGETPLATFORM" = "linux/amd64" ]; then \
       python3 -m pip install \
         --no-cache-dir --disable-pip-version-check --extra-index-url https://download.pytorch.org/whl/cpu  \
-        torch==2.3.1+cpu .[all]; \
+        torch==2.5.1+cpu .[all]; \
     else \
       python3 -m pip install \
         --no-cache-dir --disable-pip-version-check \
diff --git a/setup.py b/setup.py
index 316f73c..de64e8a 100644
--- a/setup.py
+++ b/setup.py
@@ -60,7 +60,7 @@ extras = {
     "lightgbm": ["lightgbm>=3,<5"],
     "pytorch": [
         "requests<3",
-        "torch==2.3.1",
+        "torch==2.5.1",
         "tqdm",
         "sentence-transformers>=2.1.0,<=2.7.0",
         # sentencepiece is a required dependency for the slow tokenizers