docker-compose.yml

version: '3.9'

x-base_service: &base_service
    ports:
      - "${WEBUI_PORT:-7860}:7860"
    volumes:
      - &v1 ./data:/data
      - &v2 ./output:/output
    stop_signal: SIGKILL
    tty: true
    deploy:
      resources:
        reservations:
          devices:
              - driver: nvidia
                device_ids: ['0']
                capabilities: [compute, utility]

name: webui-docker

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

  auto: &automatic
    <<: *base_service
    profiles: ["auto"]
    build: ./services/AUTOMATIC1111
    image: emprops/auto1111:sdauto
    environment:
      - CLI_ARGS=--opt-sdp-no-mem-attention --api --port 3000 --medvram --no-half-vae

  1_5-and-2_1: &automatic
    <<: *base_service
    profiles: ["1_5-and-2_1"]
    build: 
      context: ./services/AUTOMATIC1111
      args:
        models: "1_5-and-2_1"
    image: emprops/auto1111:sdauto1_5-and-2_1
    environment:
      - CLI_ARGS=--opt-sdp-no-mem-attention --api --port 3000 --medvram --no-half-vae

  sdxl-and-refiner: &automatic
    <<: *base_service
    profiles: ["sdxl-and-refiner"]
    build: 
      context: ./services/AUTOMATIC1111
      args:
        models: "sdxl-and-refiner"
    image: emprops/auto1111:sdautosdxl-and-refiner
    environment:
      - CLI_ARGS=--opt-sdp-no-mem-attention --api --port 3000 --medvram --no-half-vae

  all: &automatic
    <<: *base_service
    profiles: ["all"]
    build: 
      context: ./services/AUTOMATIC1111
      args:
        models: "models"
        GITACCESSKEY: ${GITACCESSKEY}
    image: emprops/auto1111:sdauto-allv5
    environment:
      - CLI_ARGS=--opt-sdp-no-mem-attention --api --port 3000 --medvram --no-half-vae

  old: &automatic
    <<: *base_service
    profiles: ["old"]
    build: 
      context: ./services/AUTOMATIC1111
      args:
        models: "models"
        GITACCESSKEY: ${GITACCESSKEY}
    image: emprops/auto1111:sdauto-oldv1
    environment:
      - CLI_ARGS=--opt-sdp-no-mem-attention --api --port 3000 --medvram --no-half-vae

  auto-cpu:
    <<: *automatic
    profiles: ["auto-cpu"]
    deploy: {}
    environment:
      - CLI_ARGS=--no-half --precision full --allow-code --enable-insecure-extension-access --api

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

  # invoke-cpu:
  #   <<: *invoke
  #   profiles: ["invoke-cpu"]
  #   environment:
  #     - PRELOAD=true
  #     - CLI_ARGS=--always_use_cpu

  comfy: &comfy
    <<: *base_service
    profiles: ["comfy"]
    build: 
      context: ./services/comfy
      args:
        models: "models"
        GITACCESSKEY: ${GITACCESSKEY}
    image: emprops/comfy:v1
    environment:
      - CLI_ARGS=

  comfy-cpu:
    <<: *comfy
    profiles: ["comfy-cpu"]
    deploy: {}
    environment:
      - CLI_ARGS=--cpu