unclemusclez commited on
Commit
bcfd37f
·
verified ·
1 Parent(s): 3871e58

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -50,8 +50,8 @@ ENV PYTHONUNBUFFERED=1 \
50
  LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib64:${LD_LIBRARY_PATH} \
51
  NVIDIA_DRIVER_CAPABILITIES=compute,utility \
52
  NVIDIA_VISIBLE_DEVICES=all \
53
- OLLAMA_HOST=0.0.0.0:0
 
54
 
55
  WORKDIR ${APPDIR}
56
- # RUN ollama serve & echo $! > ${HOME}/ollama.pid & sleep 5
57
  ENTRYPOINT python app.py
 
50
  LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib64:${LD_LIBRARY_PATH} \
51
  NVIDIA_DRIVER_CAPABILITIES=compute,utility \
52
  NVIDIA_VISIBLE_DEVICES=all \
53
+ OLLAMA_MODELS=${APPDIR}/.ollama/models \
54
+ OLLAMA_HOST=127.0.0.1:0
55
 
56
  WORKDIR ${APPDIR}
 
57
  ENTRYPOINT python app.py