File size: 858 Bytes
6e204a5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 | version: "3.8"
services:
floorgen-api:
build:
context: .
dockerfile: Dockerfile
container_name: floorgen-api
restart: unless-stopped
ports:
- "8000:8000"
environment:
- FLOORGEN_HOST=0.0.0.0
- FLOORGEN_PORT=8000
- FLOORGEN_DEVICE=cuda
- FLOORGEN_LOG_LEVEL=INFO
- FLOORGEN_DATA_DIR=/app/data/processed
- FLOORGEN_CHECKPOINT=/app/checkpoints/rag_diffusion.pt
volumes:
- ./data/processed:/app/data/processed:ro
- ./checkpoints:/app/checkpoints:ro
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities: [gpu]
healthcheck:
test: ["CMD", "curl", "-f", "http://localhost:8000/health"]
interval: 30s
timeout: 10s
retries: 3
start_period: 40s
|