version: '3.9'

x-base_service: &base_service
    ports:
      - "${WEBUI_PORT:-7860}:7860"
    volumes:
      - &v1 ./data:/data:Z
      - &v2 ./output:/output:Z
    security_opt:
      - label=disable
    ipc:
      - host

name: webui-podman

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

  auto-cuda: &automatic-cuda
    <<: *base_service
    profiles: ["auto-cuda"]
    build: ./services/AUTOMATIC1111-cuda
    image: sd-auto:193
    devices:
      - nvidia.com/gpu=all
    environment:
      # - CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api
      - CLI_ARGS=--allow-code --enable-insecure-extension-access --api

  auto-rocm: &automatic-rocm
    <<: *base_service
    profiles: ["auto-rocm"]
    build: ./services/AUTOMATIC1111-rocm
    image: sd-auto:193
    devices:
      - /dev/dri
      - /dev/kfd
    environment:
      # - CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api
      - CLI_ARGS=--allow-code --enable-insecure-extension-access --api


  # 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
  #     - CLI_ARGS=

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

  comfy-cuda: &comfy-cuda
    <<: *base_service
    profiles: ["comfy-cuda"]
    build: ./services/comfy-cuda/
    image: sd-comfy:5eb98f00927ace00b6b3d01ed9c76b113fc4ec9f
    devices:
      - nvidia.com/gpu=all
    environment:
      - CLI_ARGS=

  comfy-rocm: &comfy-rocm
    <<: *base_service
    profiles: ["comfy-rocm"]
    build: ./services/comfy-rocm/
    image: sd-comfy:5eb98f00927ace00b6b3d01ed9c76b113fc4ec9f
    devices:
      - /dev/dri
      - /dev/kfd
    environment:
      - CLI_ARGS=

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