compose.yaml

name: melody-engine

services:
  backend:
    build:
      context: .
      dockerfile: Dockerfile
      args:
        REVISION: "${REVISION}"
    image: localhost/melody-engine:${REVISION}
    container_name: melody-engine
    restart: unless-stopped
    environment:
      - TF_GPU_ALLOCATOR=cuda_malloc_async
    volumes:
      - ./outputs:/opt/app/outputs:rw,Z
      - ~/.cache/melody-engine:/magenta-realtime/cache:rw,Z
    ports:
      - "8080:8080"
    networks:
      - net
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]

networks:
  net:
    driver: bridge