raystermomo commited on
Commit
aa7c4b4
·
1 Parent(s): fe6e7af

Bump torch to 2.6 (transformers requires it for torch.load, CVE-2025-32434)

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +3 -1
Dockerfile CHANGED
@@ -23,7 +23,7 @@ WORKDIR /home/user/app
23
 
24
  # Install the CPU build of torch explicitly (the default PyPI wheel is CUDA and huge).
25
  RUN pip install --upgrade pip && \
26
- pip install --user torch==2.5.1 --index-url https://download.pytorch.org/whl/cpu
27
 
28
  COPY --chown=user requirements.txt .
29
  RUN pip install --user -r requirements.txt
 
23
 
24
  # Install the CPU build of torch explicitly (the default PyPI wheel is CUDA and huge).
25
  RUN pip install --upgrade pip && \
26
+ pip install --user torch==2.6.0 --index-url https://download.pytorch.org/whl/cpu
27
 
28
  COPY --chown=user requirements.txt .
29
  RUN pip install --user -r requirements.txt
requirements.txt CHANGED
@@ -6,7 +6,9 @@ anthropic>=0.39
6
  python-dotenv>=1.0
7
 
8
  # Local ML models (LINE persona analysis + history summarization)
9
- torch>=2.4,<2.7
 
 
10
  transformers>=4.44,<5
11
  sentencepiece>=0.1.99
12
  numpy>=1.26
 
6
  python-dotenv>=1.0
7
 
8
  # Local ML models (LINE persona analysis + history summarization)
9
+ # torch >= 2.6 is required: recent transformers refuses torch.load on .bin
10
+ # weights below 2.6 (CVE-2025-32434).
11
+ torch>=2.6,<2.8
12
  transformers>=4.44,<5
13
  sentencepiece>=0.1.99
14
  numpy>=1.26