docker-compose.yml

services:
  stable-diffusion-comfyui:
    image: rocm/pytorch:latest
    ports:
      - "80:80"
    environment:
      DOCKER_INSTANCE: comfyui
    restart: unless-stopped
    command: ["/conf/startup-comfyui.sh"]
    devices:
      - "/dev/kfd:/dev/kfd"
      - "/dev/dri:/dev/dri"
    extra_hosts:
      - "host.docker.internal:host-gateway"
    cap_add:
      - SYS_PTRACE
    security_opt:
      - seccomp=unconfined
    volumes:
      - .:/workspace
      - ./data/home:/root
      - ./data/checkpoints:/checkpoints
      - ./data/comfyui:/comfyui
      - ./conf:/conf

  stable-diffusion-webui:
    image: rocm/pytorch:latest
    ports:
      - "81:81"
    environment:
      DOCKER_INSTANCE: webui
    restart: unless-stopped
    command: ["/conf/startup-webui.sh"]
    devices:
      - "/dev/kfd:/dev/kfd"
      - "/dev/dri:/dev/dri"
    extra_hosts:
      - "host.docker.internal:host-gateway"
    cap_add:
      - SYS_PTRACE
    security_opt:
      - seccomp=unconfined
    volumes:
      - .:/workspace
      - ./data/home:/root
      - ./data/checkpoints:/checkpoints
      - ./data/sd-webui:/sd-webui
      - ./conf:/conf