binary1ne commited on
Commit
306df80
·
verified ·
1 Parent(s): 9fd8915

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +11 -1
Dockerfile CHANGED
@@ -14,4 +14,14 @@ RUN pip list
14
 
15
  RUN vllm -v
16
 
17
- CMD ["vllm", "serve", "unsloth/Llama-3.2-3B-bnb-4bit", "--host", "0.0.0.0", "--port", "7860", "--trust-remote-code", "--device", "cpu"]
 
 
 
 
 
 
 
 
 
 
 
14
 
15
  RUN vllm -v
16
 
17
+ RUN git lfs install && \
18
+ git clone https://huggingface.co/unsloth/Llama-3.2-3B-bnb-4bit /workspace/model
19
+
20
+ CMD ["python", "-m", "vllm.entrypoints.openai.api_server", \
21
+ "--model", "/workspace/model", \
22
+ "--host", "0.0.0.0", \
23
+ "--port", "7860", \
24
+ "--trust-remote-code", \
25
+ "--device", "cpu"]
26
+
27
+ # CMD ["vllm", "serve", "unsloth/Llama-3.2-3B-bnb-4bit", "--host", "0.0.0.0", "--port", "7860", "--trust-remote-code", "--device", "cpu"]