services:
  stable-diffusion-webui-forge:
    build:
      context: ./stable-diffusion-webui-forge
      args: &buid_args
        PYTHON_VERSION: "3.10"
        ROCM_VERSION: "6.1"
    restart: unless-stopped
    environment:
      - HSA_OVERRIDE_GFX_VERSION=11.0.0
    ports:
     - 7860:7860
    devices:
      - /dev/kfd
      - /dev/dri
    volumes:
      - ./shared/loras:/app/models/Lora
      - ./shared/checkpoints:/app/models/Stable-diffusion
      - ./shared/vae/:/app/models/VAE
      - ./shared/embeddings:/app/embeddings
      - ./stable-diffusion-webui-forge/extensions:/app/extensions
      - ./stable-diffusion-webui-forge/repositories:/app/repositories
      - ./stable-diffusion-webui-forge/outputs:/app/outputs
      - ./stable-diffusion-webui-forge/favorites:/app/log/images
    group_add:
      - video
    ipc: host
    cap_add:
      - SYS_PTRACE
    security_opt:
      - seccomp=unconfined
  comfyui:
    build:
      context: ./ComfyUI
      args:
        <<: *buid_args
    command: ["--fp16-vae"]
    restart: unless-stopped
    environment:
      - HSA_OVERRIDE_GFX_VERSION=11.0.0
    ports:
      - 8188:8188
    devices:
      - /dev/kfd
      - /dev/dri
    volumes:
      - ./shared/loras:/app/models/loras
      - ./shared/checkpoints:/app/models/checkpoints
      - ./shared/vae/:/app/models/vae
      - ./shared/embeddings:/app/models/embeddings
      - ./ComfyUI/custom_nodes:/app/custom_nodes
      - ./ComfyUI/outputs:/app/output
      - ./ComfyUI/inputs:/app/input
    group_add:
      - video
    ipc: host
    cap_add:
      - SYS_PTRACE
    security_opt:
      - seccomp=unconfined