version: '3.8'

services:
  dia-api:
    build: .
    container_name: dia-api-server
    ports:
      - "8000:8000"
    volumes:
      - hf_cache:/root/.cache/huggingface
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    environment:
      - MODEL_ID=nari-labs/Dia-1.6B
      - PORT=8000

volumes:
  hf_cache: