harsh-dev commited on
Commit
7599817
·
unverified ·
1 Parent(s): 67ef30c

fix: upgrade torch to 2.5.1 for NumPy 2.x and Accelerate support

Browse files
Files changed (1) hide show
  1. 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.2.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.