From c3211fbce92ce330605de84edaabfc18ddcbaa3b Mon Sep 17 00:00:00 2001 From: davidmezzetti <561939+davidmezzetti@users.noreply.github.com> Date: Thu, 4 Jun 2026 10:57:39 -0400 Subject: [PATCH] Bump Torch version --- docker/base/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/base/Dockerfile b/docker/base/Dockerfile index 97f53ba2..4fd60a05 100644 --- a/docker/base/Dockerfile +++ b/docker/base/Dockerfile @@ -27,7 +27,7 @@ RUN \ # Install txtai project and dependencies ln -s /usr/bin/python${PYTHON_VERSION} /usr/bin/python && \ python -m pip install --no-cache-dir -U pip wheel setuptools && \ - if [ -z ${GPU} ]; then pip install --no-cache-dir torch==2.11.0+cpu torchvision==0.26.0+cpu -f https://download.pytorch.org/whl/torch -f https://download.pytorch.org/whl/torchvision; fi && \ + if [ -z ${GPU} ]; then pip install --no-cache-dir torch==2.12.0+cpu torchvision==0.27.0+cpu -f https://download.pytorch.org/whl/torch -f https://download.pytorch.org/whl/torchvision; fi && \ python -m pip install --no-cache-dir txtai${COMPONENTS} && \ python -c "import sys, importlib.util as util; 1 if util.find_spec('nltk') else sys.exit(); import nltk; nltk.download(['punkt', 'punkt_tab', 'averaged_perceptron_tagger_eng'])" && \ \