services:
  stable-diffusion-webui:
    build:
      context: .
      dockerfile: Dockerfile
    deploy:
      resources:
        limits:
          memory: 16G
          cpus: 16
    container_name: stable-diffusion-webui
    restart: unless-stopped
    environment:
      - HSA_OVERRIDE_GFX_VERSION=11.0.0
      - HIP_VISIBLE_DEVICES=0
      - PYTORCH_ROCM_ARCH=gfx1100
      - CLI_ARGS=--api --listen --port 7860 --opt-sdp-attention --no-half-vae --skip-torch-cuda-test
    ports:
      - "7860:7860"
    volumes:
      - ./models:/app/models
      - ./outputs:/app/outputs
      - ./extensions:/app/extensions
    devices:
      - "/dev/kfd:/dev/kfd"
      - "/dev/dri:/dev/dri"