version: '3.9'

x-base_service: &base_service
    ports:
      - "7860:7860"
    volumes: &base_volumes
      - ./cache:/cache
      - ./output:/output
      - ./models:/models
    deploy:
      resources:
        reservations:
          devices:
              - driver: nvidia
                device_ids: ['0']
                capabilities: [gpu]

name: webui-docker

services:
  hlky:
    <<: *base_service
    profiles: ["hlky"]
    build: ./services/hlky/
    environment:
      - CLI_ARGS=--extra-models-cpu --optimized-turbo

  automatic1111: &automatic
    <<: *base_service
    profiles: ["auto"]
    build: ./services/AUTOMATIC1111
    environment:
      - CLI_ARGS=--medvram --opt-split-attention

  automatic1111-cpu:
    <<: *automatic
    profiles: ["auto-cpu"]
    deploy: {}
    environment:
      - CLI_ARGS=--no-half --precision full

  lstein:
    <<: *base_service
    profiles: ["lstein"]
    build: ./services/lstein/