From a7980f846a3526acb97399aae75bf2cc62329253 Mon Sep 17 00:00:00 2001 From: Quentin Pradet Date: Tue, 28 Oct 2025 12:02:04 +0400 Subject: [PATCH 1/2] Upgrade PyTorch to 2.7.1 --- Dockerfile | 2 +- Dockerfile.wolfi | 2 +- setup.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/Dockerfile b/Dockerfile index 4119f2e4..cbee0280 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.5.1+cpu .[all]; \ + torch==2.7.1+cpu .[all]; \ else \ python3 -m pip install \ --no-cache-dir --disable-pip-version-check \ diff --git a/Dockerfile.wolfi b/Dockerfile.wolfi index d54be8fd..d83dec38 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.5.1+cpu .[all]; \ + torch==2.7.1+cpu .[all]; \ else \ python3 -m pip install \ --no-cache-dir --disable-pip-version-check \ diff --git a/setup.py b/setup.py index f17023d8..15455f9a 100644 --- a/setup.py +++ b/setup.py @@ -60,7 +60,7 @@ "lightgbm": ["lightgbm>=3,<5"], "pytorch": [ "requests<3", - "torch==2.5.1", + "torch==2.7.1", "tqdm", "sentence-transformers>=5.0.0,<6.0.0", # sentencepiece is a required dependency for the slow tokenizers From 3cb89b1031361083a751842d456ee852618fe7c7 Mon Sep 17 00:00:00 2001 From: Quentin Pradet Date: Wed, 29 Oct 2025 12:16:53 +0400 Subject: [PATCH 2/2] Bump transformers dependency --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 87c02581..c23dc810 100644 --- a/setup.py +++ b/setup.py @@ -65,7 +65,7 @@ "sentence-transformers>=5.0.0,<6.0.0", # sentencepiece is a required dependency for the slow tokenizers # https://huggingface.co/transformers/v4.4.2/migration.html#sentencepiece-is-removed-from-the-required-dependencies - "transformers[sentencepiece]>=4.47.0,<4.50.3", + "transformers[sentencepiece]>=4.0.0,<5.0.0", ], } extras["all"] = list({dep for deps in extras.values() for dep in deps})