Spaces:
Runtime error
Runtime error
Commit ·
aa7c4b4
1
Parent(s): fe6e7af
Bump torch to 2.6 (transformers requires it for torch.load, CVE-2025-32434)
Browse files- Dockerfile +1 -1
- 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.
|
| 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.
|
|
|
|
|
|
|
| 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
|