version: '3.4'
services:
  auto-sd-docker:
    container_name: auto-sd-docker
    image: auto-sd-docker:latest
    build:
      context: .
    restart: "no"
    ports:
      - "7860:7860"
    volumes:
      - ./:/repo # The path the repo, will be cloned into if it does not already exist
      # Paths to existing models. Should bind to a subdirectory of the repo folder for that type with examples below
      # - "/path/to/models:/repo/models/Stable-diffusion/remote"
      # - "/path/to/embeddings:/repo/embeddings/remote"
      # - "/path/to/VAE:/repo/models/VAE/remote"
      # - "/path/to/Lora:/repo/models/Lora/remote"
    deploy:
      resources:
        reservations:
          devices:
              - driver: nvidia
                device_ids: ['0']
                capabilities: [gpu]
    healthcheck:
      test: "wget --no-verbose --tries=1 --spider http://localhost:7860 || exit 1"
      interval: 60s
      timeout: 10s
      retries: 5
      start_period: 420s