Hugging Face's logo

Spaces:
Dovakiins
/
qwerrwe
Build error

App Files Community
winglian commited on
Commit
ece46b2
•
1 Parent(s): 92d800a

pip install packaging dep

Browse files
Files changed (1) hide show
  1. docker/Dockerfile-base +2 -1
docker/Dockerfile-base CHANGED
@@ -27,7 +27,8 @@ ENV PATH="/root/miniconda3/envs/py${PYTHON_VERSION}/bin:${PATH}"
27
 
28
  WORKDIR /workspace
29
 
30
- RUN python3 -m pip install --no-cache-dir -U torch==${PYTORCH} torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/$CUDA
 
31
 
32
 
33
  FROM base-builder AS flash-attn-builder
 
27
 
28
  WORKDIR /workspace
29
 
30
+ RUN python3 -m pip install --upgrade pip && pip3 install packaging && \
31
+ python3 -m pip install --no-cache-dir -U torch==${PYTORCH} torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/$CUDA
32
 
33
 
34
  FROM base-builder AS flash-attn-builder