version: '3.9'

x-base_service: &base_service
    ports:
      - "7860:7860"
      - "7861:7861"
    volumes:
      - &v1 ./data:/data
      - &v2 ./output:/output
    deploy:
      resources:
        reservations:
          devices:
              - driver: nvidia
                device_ids: ['0,1']
                capabilities: [gpu]

name: webui-docker

services:
  download:
    build: ./services/download/
    profiles: ["download"]
    volumes:
      - *v1

  auto0: &automatic0
    <<: *base_service
    profiles: ["auto0"]
    build: ./services/AUTOMATIC1111_0
    image: sd-auto0:33
    environment:
      - CLI_ARGS=--allow-code --xformers --enable-insecure-extension-access --api
      
  auto1: &automatic1
    <<: *base_service
    profiles: ["auto1"]
    build: ./services/AUTOMATIC1111_1
    image: sd-auto1:33
    environment:
      - CLI_ARGS=--allow-code --xformers --enable-insecure-extension-access --api

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

  invoke:
    <<: *base_service
    profiles: ["invoke"]
    build: ./services/invoke/
    image: sd-invoke:17
    environment:
      - PRELOAD=true
      - CLI_ARGS=


  sygil: &sygil
    <<: *base_service
    profiles: ["sygil"]
    build: ./services/sygil/
    image: sd-sygil:16
    environment:
      - CLI_ARGS=--optimized-turbo
      - USE_STREAMLIT=0

  sygil-sl:
    <<: *sygil
    profiles: ["sygil-sl"]
    environment:
      - USE_STREAMLIT=1