ubix commited on
Commit
634a45b
·
verified ·
1 Parent(s): 6dde48b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -9
Dockerfile CHANGED
@@ -2,14 +2,7 @@ FROM ghcr.io/ggml-org/llama.cpp:full
2
 
3
  RUN apt update && apt install wget -y
4
 
5
- # Download model
6
  RUN wget "https://huggingface.co/unsloth/gemma-3-1b-it-GGUF/resolve/main/gemma-3-1b-it-Q4_K_M.gguf" -O /model.gguf
7
 
8
- # API-optimized settings: larger context, GPU offload, flash attention
9
- CMD ["--server", "-m", "/model.gguf", "--port", "7860", "--host", "0.0.0.0",
10
- "-c", "4096", # Context size: 4K tokens
11
- "-n", "1024", # Max generation: 1K tokens
12
- "-ngl", "999", # Offload all layers to GPU (if available)
13
- "-fa", # Flash Attention for speed
14
- "--chat-template", "gemma", # Proper chat formatting
15
- "--timeout", "300"] # 5 min timeout for long requests
 
2
 
3
  RUN apt update && apt install wget -y
4
 
 
5
  RUN wget "https://huggingface.co/unsloth/gemma-3-1b-it-GGUF/resolve/main/gemma-3-1b-it-Q4_K_M.gguf" -O /model.gguf
6
 
7
+ # Single line CMD - no comments inside
8
+ CMD ["--server", "-m", "/model.gguf", "--port", "7860", "--host", "0.0.0.0", "-c", "4096", "-n", "1024", "-ngl", "999", "-fa", "--chat-template", "gemma"]