ilhooq commited on
Commit
435f6db
·
1 Parent(s): 45c1114
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM ghcr.io/ggerganov/llama.cpp:server
2
 
3
  RUN apt update && apt install -y curl
4
 
@@ -8,4 +8,4 @@ RUN curl -L https://huggingface.co/TheBloke/deepseek-coder-6.7B-instruct-GGUF/re
8
 
9
  ENTRYPOINT [ "/server" ]
10
 
11
- CMD [ "--host", "0.0.0.0", "--port", "7860", "--model", "/models/deepseek-coder-6.7b-instruct.Q6_K.gguf", "-c", "4096", "--chat-template", "deepseek" ]
 
1
+ FROM ghcr.io/ggerganov/llama.cpp:server-cuda
2
 
3
  RUN apt update && apt install -y curl
4
 
 
8
 
9
  ENTRYPOINT [ "/server" ]
10
 
11
+ CMD [ "-ngl", "32" "--host", "0.0.0.0", "--port", "7860", "--model", "/models/deepseek-coder-6.7b-instruct.Q6_K.gguf", "-c", "4096", "--chat-template", "deepseek" ]