Spaces:
Running
Running
Update Dockerfile
Browse files- Dockerfile +4 -4
Dockerfile
CHANGED
|
@@ -2,11 +2,11 @@ FROM ghcr.io/ggml-org/llama.cpp:server
|
|
| 2 |
|
| 3 |
RUN apt update && apt install wget -y && rm -rf /var/lib/apt/lists/*
|
| 4 |
|
| 5 |
-
# THE MAIN MODEL:
|
| 6 |
-
|
| 7 |
-
RUN wget "https://huggingface.co/bartowski/gemma-2-2b-it-GGUF/resolve/main/gemma-2-2b-it-Q4_K_M.gguf" -O /model.gguf
|
| 8 |
|
| 9 |
# Run the server natively (Text-Only Mode)
|
|
|
|
| 10 |
CMD ["-m", "/model.gguf", \
|
| 11 |
"--port", "7860", "--host", "0.0.0.0", \
|
| 12 |
-
"-t", "2", "--mlock", "-c", "
|
|
|
|
| 2 |
|
| 3 |
RUN apt update && apt install wget -y && rm -rf /var/lib/apt/lists/*
|
| 4 |
|
| 5 |
+
# 1. THE MAIN MODEL: Using Q4_K_M for speed on 2vCPUs
|
| 6 |
+
RUN wget "https://huggingface.co/unsloth/gemma-3n-E2B-it-GGUF/resolve/main/gemma-3n-E2B-it-Q4_K_M.gguf" -O /model.gguf
|
|
|
|
| 7 |
|
| 8 |
# Run the server natively (Text-Only Mode)
|
| 9 |
+
# -c 4096 is the safety limit to prevent RAM crashes
|
| 10 |
CMD ["-m", "/model.gguf", \
|
| 11 |
"--port", "7860", "--host", "0.0.0.0", \
|
| 12 |
+
"-t", "2", "--mlock", "-c", "8196"]
|