From 878cde61263a154f6d247672ea3e4f6bc8c89ced Mon Sep 17 00:00:00 2001 From: David Kyle Date: Wed, 30 Apr 2025 10:57:45 +0100 Subject: [PATCH] Upgrade PyTorch to 2.5.1 (#785) PyTorch was upgraded to 2.5.1 in ml-cpp on the 8.18 and 9.0 branches in elastic/ml-cpp#2800 --- Dockerfile | 2 +- Dockerfile.wolfi | 2 +- setup.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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