seanpedrickcase commited on
Commit
cd6a3e0
·
1 Parent(s): 22ca76e

Added NUMBA_CACHE_DIR to Docker environmental variables

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -34,6 +34,7 @@ RUN mkdir -p /home/user/.cache/huggingface/hub && chown -R user:user /home/user/
34
  RUN mkdir -p /home/user/.cache/matplotlib && chown -R user:user /home/user/.cache/matplotlib
35
  RUN mkdir -p /home/user/app/model/rep && chown -R user:user /home/user/app/model/rep
36
  RUN mkdir -p /home/user/app/model/embed && chown -R user:user /home/user/app/model/embed
 
37
 
38
  # Download the quantised phi model directly with curl. Changed at it is so big - not loaded
39
  #RUN curl -L -o /home/user/app/model/rep/Phi-3.1-mini-128k-instruct-Q4_K_M.gguf https://huggingface.co/bartowski/Phi-3.1-mini-128k-instruct-GGUF/tree/main/Phi-3.1-mini-128k-instruct-Q4_K_M.gguf
@@ -66,6 +67,7 @@ ENV HOME=/home/user \
66
  AWS_STS_REGIONAL_ENDPOINT=regional \
67
  GRADIO_OUTPUT_FOLDER='output/' \
68
  #GRADIO_ROOT_PATH=/data-text-search \
 
69
  SYSTEM=spaces
70
 
71
  # Set the working directory to the user's home directory
 
34
  RUN mkdir -p /home/user/.cache/matplotlib && chown -R user:user /home/user/.cache/matplotlib
35
  RUN mkdir -p /home/user/app/model/rep && chown -R user:user /home/user/app/model/rep
36
  RUN mkdir -p /home/user/app/model/embed && chown -R user:user /home/user/app/model/embed
37
+ RUN mkdir -p /home/user/app/cache && chown -R user:user /home/user/app/cache
38
 
39
  # Download the quantised phi model directly with curl. Changed at it is so big - not loaded
40
  #RUN curl -L -o /home/user/app/model/rep/Phi-3.1-mini-128k-instruct-Q4_K_M.gguf https://huggingface.co/bartowski/Phi-3.1-mini-128k-instruct-GGUF/tree/main/Phi-3.1-mini-128k-instruct-Q4_K_M.gguf
 
67
  AWS_STS_REGIONAL_ENDPOINT=regional \
68
  GRADIO_OUTPUT_FOLDER='output/' \
69
  #GRADIO_ROOT_PATH=/data-text-search \
70
+ NUMBA_CACHE_DIR=/home/user/app/cache \
71
  SYSTEM=spaces
72
 
73
  # Set the working directory to the user's home directory