Update Dockerfile

Dockerfile

@@ -9,7 +9,10 @@ RUN mkdir -p $HF_HOME && chmod -R 777 $HF_HOME
 RUN mkdir -p /workspace/models && chmod -R 777 /workspace/models
 
 # Install git & git-lfs
-RUN apt-get update && apt-get install -y git git-lfs && \
+RUN apt-get update && \
+    DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
+    util-linux numactl procps curl ca-certificates \
+    git git-lfs && \
     git lfs install && \
     apt-get clean && rm -rf /var/lib/apt/lists/*
 
@@ -24,7 +27,7 @@ RUN pip list
 
 RUN vllm -v
 
-RUN vllm serve unsloth/Llama-3.2-3B-bnb-4bit
+# RUN vllm serve unsloth/Llama-3.2-3B-bnb-4bit
 
 # RUN git lfs install && \
 #     git clone https://huggingface.co/unsloth/Llama-3.2-1B-bnb-4bit /workspace/models
@@ -36,4 +39,4 @@ RUN vllm serve unsloth/Llama-3.2-3B-bnb-4bit
 #      "--trust-remote-code", \
 #      "--device", "cpu"]
 
-# CMD ["vllm", "serve", "unsloth/Llama-3.2-1B-bnb-4bit", "--host", "0.0.0.0", "--port", "7860", "--trust-remote-code", "--device", "cpu"]
+CMD ["vllm", "serve", "unsloth/Llama-3.2-1B-bnb-4bit", "--host", "0.0.0.0", "--port", "7860", "--trust-remote-code", "--device", "cpu"]