services:
  stable-diffusion:
    container_name: stable-diffusion
    image: ghcr.io/ai-dock/stable-diffusion-webui-forge:latest-cuda
    restart: unless-stopped
    ports:
      - 7860:7860
    environment:
      - FORGE_ARGS=--xformers --api
      - AUTO_UPDATE=false
      - WEB_ENABLE_AUTH=false
      - TZ=America/New_York
      # Optional - for gated models
      # - HF_TOKEN=your_token
      # - CIVITAI_TOKEN=your_token
    volumes:
      - /srv/stable-diffusion/workspace:/workspace
      # Optional: custom provisioning script
      # - /srv/stable-diffusion/provisioning.sh:/opt/ai-dock/bin/provisioning.sh
    runtime: nvidia
    networks:
      - npm-network

networks:
  npm-network:
    external: true