gemma-4-api-poc / Dockerfile
NS-Genai's picture
Create Dockerfile
cc60fed verified
Raw
History Blame Contribute Delete
711 Bytes
# Use the official vLLM image as our base
FROM vllm/vllm-openai:latest
# Set environment variables so vLLM binds to the correct HF port
ENV HOST="0.0.0.0"
ENV PORT="7860"
# HF Spaces runs as a non-root user. We need to tell vLLM to download
# the model into a temporary folder where it has write permissions.
ENV HF_HOME="/tmp/.cache/huggingface"
RUN mkdir -p /tmp/.cache/huggingface && chmod 777 /tmp/.cache/huggingface
# Start the OpenAI-compatible vLLM server
ENTRYPOINT ["python3", "-m", "vllm.entrypoints.openai.api_server"]
# Pass the model configurations
CMD ["--model", "google/gemma-4-E4B-it", "--max-model-len", "4096", "--dtype", "half", "--enforce-eager", "--host", "0.0.0.0", "--port", "7860"]