services:
  api:
    build:
      context: .
      args:
        VARIANT: gpu
    ports:
      - "8000:8000"
    volumes:
      - ./data/uploads:/data/uploads
      - ./data/outputs:/data/outputs
      - ./data/models:/data/models
      - ./data/temp:/data/temp
      - ./data/jobs:/data/jobs
    environment:
      - RSR_EXECUTION_PROVIDERS=["cuda"]
      - RSR_EXECUTION_THREAD_COUNT=8
      - RSR_TILE_SIZE=400
      - RSR_LOG_LEVEL=info
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    restart: unless-stopped