Hugging Face's logo

Spaces:
RWKV-Red-Team
/
RWKV-LatestSpace
Running on T4

App Files Community
1
sparkleman commited on
Commit
9b15eee
·
1 Parent(s): 08af64e

UPDATE: HF_HOME persist

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -46,7 +46,8 @@ ENV HOME=/home/user \
46
  PATH=/usr/local/cuda/bin:/home/user/.local/bin:$PATH \
47
  LD_LIBRARY_PATH="/usr/local/cuda/lib64:${LD_LIBRARY_PATH}" \
48
  CXX=/usr/bin/g++ \
49
- TORCH_CUDA_ARCH_LIST="7.5"
 
50
  WORKDIR $HOME/app
51
 
52
  COPY --chown=user . $HOME/app
 
46
  PATH=/usr/local/cuda/bin:/home/user/.local/bin:$PATH \
47
  LD_LIBRARY_PATH="/usr/local/cuda/lib64:${LD_LIBRARY_PATH}" \
48
  CXX=/usr/bin/g++ \
49
+ TORCH_CUDA_ARCH_LIST="7.5" \
50
+ HF_HOME="/data/.huggingface"
51
  WORKDIR $HOME/app
52
 
53
  COPY --chown=user . $HOME/app