H-Liu1997 commited on
Commit
c62140b
·
1 Parent(s): a4f8eb3

fix: split pip install to avoid index-url conflict for non-torch packages

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -5,9 +5,9 @@ RUN apt-get update && apt-get install -y \
5
  && rm -rf /var/lib/apt/lists/*
6
 
7
  RUN pip3 install --no-cache-dir \
8
- flask flask-cors \
9
- torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 \
10
- numpy transformers accelerate sentencepiece protobuf
11
 
12
  # Create non-root user (HF Spaces runs as uid 1000)
13
  RUN useradd -m -u 1000 user
 
5
  && rm -rf /var/lib/apt/lists/*
6
 
7
  RUN pip3 install --no-cache-dir \
8
+ torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 && \
9
+ pip3 install --no-cache-dir \
10
+ flask flask-cors numpy transformers accelerate sentencepiece protobuf
11
 
12
  # Create non-root user (HF Spaces runs as uid 1000)
13
  RUN useradd -m -u 1000 user