preu920 commited on
Commit
640d394
·
1 Parent(s): f12ab63

Fix HF CPU runtime by forcing CPU-only torch/torchaudio wheels

Browse files
Files changed (2) hide show
  1. Dockerfile +5 -2
  2. requirements.txt +0 -1
Dockerfile CHANGED
@@ -24,9 +24,12 @@ ENV CUDA_VISIBLE_DEVICES=""
24
  # So Render logs show startup output immediately
25
  ENV PYTHONUNBUFFERED=1
26
 
27
- # Install Python deps (torch from PyPI works on CPU when CUDA_VISIBLE_DEVICES is unset)
 
28
  COPY requirements.txt .
29
- RUN pip install --no-cache-dir -r requirements.txt
 
 
30
 
31
  COPY . .
32
 
 
24
  # So Render logs show startup output immediately
25
  ENV PYTHONUNBUFFERED=1
26
 
27
+ # Install CPU-only torch stack first, then project deps.
28
+ # This avoids CUDA-linked wheels that fail with libcudart.so.* on CPU Spaces.
29
  COPY requirements.txt .
30
+ RUN pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cpu \
31
+ torch==2.8.0 torchaudio==2.8.0 \
32
+ && pip install --no-cache-dir -r requirements.txt
33
 
34
  COPY . .
35
 
requirements.txt CHANGED
@@ -3,7 +3,6 @@ neucodec>=0.0.4
3
  numpy==2.2.6
4
  phonemizer==3.3.0
5
  soundfile==0.13.1
6
- torch==2.8.0
7
  transformers==4.56.1
8
  resemble-perth==1.0.1
9
  gradio>=5.0
 
3
  numpy==2.2.6
4
  phonemizer==3.3.0
5
  soundfile==0.13.1
 
6
  transformers==4.56.1
7
  resemble-perth==1.0.1
8
  gradio>=5.0