version: '2'

services:
  stable-diffusion:
    build: ./
    image: stable-diffusion
    container_name: stable-diffusion
    restart: "no"
    devices:
      - /dev/nvidia0
      - /dev/nvidiactl
      - /dev/nvidia-uvm
      - /dev/nvidia-uvm-tools
    volumes:
      - ./output:/output