ngxson HF staff commited on
Commit
0bc11f6
·
verified ·
1 Parent(s): e7b6ed0

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -39,5 +39,5 @@ RUN sed -i 's/-lcublas /-lcublas_static /g' Makefile && \
39
 
40
  RUN npm i -g http-server
41
 
42
- CMD ["bash", "-c", "(http-server -p 7860 -c-1)& GGML_CUDA=1 LLAMA_CURL=1 make llama-server -j && ldd llama-server && sleep 999999999"]
43
 
 
39
 
40
  RUN npm i -g http-server
41
 
42
+ CMD ["bash", "-c", "(http-server -p 7860 -c-1)& CUDA_DOCKER_ARCH=compute_80 GGML_CUDA=1 LLAMA_CURL=1 make llama-server -j && ldd llama-server && sleep 999999999"]
43