Use GPU
Browse files- 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" ]
|