Docker compose vllm (#1821)

This commit is contained in:
Kelvin He
2024-10-07 17:20:41 -07:00
committed by GitHub
parent 5501f6d92f
commit aac047c219

35
docker-compose-vllm.yaml Normal file
View File

@@ -0,0 +1,35 @@
version: '3.8'
services:
letta:
image: lettaai/letta:latest
ports:
- "8083:8083"
environment:
- LETTA_LLM_ENDPOINT=http://vllm:8000
- LETTA_LLM_ENDPOINT_TYPE=vllm
- LETTA_LLM_MODEL=${LETTA_LLM_MODEL} # Replace with your model
- LETTA_LLM_CONTEXT_WINDOW=8192
depends_on:
- vllm
vllm:
image: vllm/vllm-openai:latest
runtime: nvidia
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities: [gpu]
environment:
- HUGGING_FACE_HUB_TOKEN=${HUGGING_FACE_HUB_TOKEN}
volumes:
- ~/.cache/huggingface:/root/.cache/huggingface
ports:
- "8000:8000"
command: >
--model ${LETTA_LLM_MODEL} --max_model_len=8000
# Replace with your model
ipc: host