version: "3"
services:
  stable-diffusion:
    build: .
    environment:
      # for 5700xt compat
      # ref: https://old.reddit.com/r/StableDiffusion/comments/ww436j/howto_stable_diffusion_on_an_amd_gpu/imcpmo8/
      - HSA_OVERRIDE_GFX_VERSION=10.3.0
      - PYTORCH_HIP_ALLOC_CONF=garbage_collection_threshold:0.9,max_split_size_mb:512
      - COMMANDLINE_ARGS=--listen --theme=dark --precision full --no-half
      # - COMMANDLINE_ARGS=--listen --theme=dark --precision full --no-half --medvram
      # - COMMANDLINE_ARGS=--listen --theme=dark --use-cpu all --precision full --no-half --skip-torch-cuda-test
    ports:
     - 7860:7860
    devices:
      - /dev/kfd
      - /dev/dri
    volumes:
      - ./shared:/shared
    working_dir: /shared
    group_add:
      - video
    ipc: host
    cap_add:
      - SYS_PTRACE
    security_opt:
      - seccomp=unconfined