Docker-compose-build.yml

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: sd-auto:80
    environment:
      - CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api

    auto-rocm:
    <<: *base_service
    profiles: ["auto-rocm"]
    build: 
      context: ./services/AUTOMATIC1111
      dockerfile: Dockerfile.rocm
    devices:
      - "/dev/kfd"
      - "/dev/dri"
    deploy: {}
    environment:
      - CLI_ARGS=--allow-code --medvram --enable-insecure-extension-access --api
      
  reforge: &reforge
    <<: *base_service
    profiles: ["reforge"]
    build: ./services/reforge
    image: sd-reforge:80
    environment:
      - CLI_ARGS=--allow-code --xformers --enable-insecure-extension-access --api --pin-shared-memory --cuda-malloc --cuda-stream

  forge: &forge
    <<: *base_service
    profiles: ["forge"]
    build: ./services/forge
    image: sd-forge:80
    environment:
      - CLI_ARGS=--allow-code --xformers --enable-insecure-extension-access --api --pin-shared-memory --cuda-malloc --cuda-stream

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

  comfy: &comfy
    <<: *base_service
    profiles: ["comfy"]
    build: ./services/comfy/
    image: sd-comfy:6
    environment:
      - CLI_ARGS=

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

  fooocus: &fooocus
    <<: *base_service
    profiles: ["fooocus"]
    build: ./services/fooocus/
    image: sd-fooocus:3
    environment:
      - CLI_ARGS=