pets123 commited on
Commit
42b28b0
·
verified ·
1 Parent(s): 4a7fccb

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -31,6 +31,7 @@ ENV PYTHONFAULTHANDLER=1 \
31
  TORCH_NVCC_FLAGS="-Xfatbin -compress-all"
32
 
33
  ENV HF_HOME=/app/.cache
 
34
 
35
  # Install system dependencies
36
  RUN apt-get update && apt-get install -y --no-install-recommends \
 
31
  TORCH_NVCC_FLAGS="-Xfatbin -compress-all"
32
 
33
  ENV HF_HOME=/app/.cache
34
+ RUN mkdir -p $HF_HOME && chmod -R 777 $HF_HOME
35
 
36
  # Install system dependencies
37
  RUN apt-get update && apt-get install -y --no-install-recommends \