version: "3.8"

services:
  stable-diffusion-webui:
    image: universonic/stable-diffusion-webui:latest
    container_name: stable-diffusion-webui
    restart: unless-stopped
    ports:
      - "${SD_PORT:-7860}:7860"
    volumes:
      - sd_data:/data
      - sd_output:/output
    environment:
      - CLI_ARGS=${CLI_ARGS:---listen --api --xformers}
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

volumes:
  sd_data:
  sd_output: