services:
  flux-api:
    image: flux-image:1.0
    container_name: flux-api
    ports:
      - "5005:5005"
    ipc: host
    ulimits:
      memlock: -1
      stack: 67108864
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
      - HF_TOKEN=${HF_TOKEN}
      - HF_HOME=/root/.cache/huggingface
      - SESSION_SECRET=${SESSION_SECRET}
    volumes:
      - /opt/huggingface_cache:/root/.cache/huggingface
      - /home/sysuser/Development/flux-api/users.json:/app/users.json
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    restart: unless-stopped