Spaces:
Sleeping
Sleeping
fix: upgrade torch to 2.5.1 for NumPy 2.x and Accelerate support
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -29,7 +29,7 @@ COPY requirements.txt .
|
|
| 29 |
# Install CPU-only PyTorch FIRST (prevents pip from pulling 2+ GB GPU wheels
|
| 30 |
# when sentence-transformers later requests torch as a dependency).
|
| 31 |
RUN pip install --no-cache-dir \
|
| 32 |
-
torch==2.
|
| 33 |
--index-url https://download.pytorch.org/whl/cpu
|
| 34 |
|
| 35 |
# Install the rest of the requirements.
|
|
|
|
| 29 |
# Install CPU-only PyTorch FIRST (prevents pip from pulling 2+ GB GPU wheels
|
| 30 |
# when sentence-transformers later requests torch as a dependency).
|
| 31 |
RUN pip install --no-cache-dir \
|
| 32 |
+
torch==2.5.1 \
|
| 33 |
--index-url https://download.pytorch.org/whl/cpu
|
| 34 |
|
| 35 |
# Install the rest of the requirements.
|