forked from ai-dock/stable-diffusion-webui
-
Notifications
You must be signed in to change notification settings - Fork 1
/
docker-compose.yaml
76 lines (68 loc) · 2.4 KB
/
docker-compose.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
version: "3.8"
# Compose file build variables set in .env
services:
supervisor:
build:
context: ./build
args:
IMAGE_BASE: ${IMAGE_BASE:-ghcr.io/ai-dock/jupyter-pytorch:latest-cuda}
tags:
- "ghcr.io/ai-dock/webui:${IMAGE_TAG:-latest-jupyter}"
image: ghcr.io/ai-dock/stable-diffusion-webui:${IMAGE_TAG:-latest-jupyter}
security_opt:
# For Rclone mount
- apparmor:unconfined
# For AMD GPU
#- seccomp:unconfined
cap_add:
# For Rclone mount
- SYS_ADMIN
devices:
# For Rclone mount
- "/dev/fuse:/dev/fuse"
# For AMD GPU
#- "/dev/kfd:/dev/kfd"
#- "/dev/dri:/dev/dri"
# For AMD GPU
#group_add:
# - video
# - render
volumes:
# For Rclone mount
- /etc/passwd:/etc/passwd:ro
- /etc/group:/etc/group:ro
- ./config/rclone:/etc/rclone
# Workspace
- ./workspace:${WORKSPACE:-/workspace/}:rshared
# Will echo to root-owned authorized_keys file;
# Avoids changing local file owner
- ./config/authorized_keys:/root/.ssh/authorized_keys_mount
#- ./config/provisioning/default.sh:/opt/ai-dock/bin/provisioning.sh
ports:
# SSH available on host machine port 2222 to avoid conflict. Change to suit
- ${SSH_PORT_HOST:-2222}:${SSH_PORT:-22}
# redirect to Cloudflare quick tunnel
- ${REDIRECTOR_PORT_HOST}:1111
# Websocket log viewer
- ${LOG_VIEWER_PORT_HOST}:1122
# ComfyUI web interface
- ${WEBUI_PORT_HOST:-7860}:${WEBUI_PORT:-7860}
# Jupyter server
- ${JUPYTER_PORT_HOST:-8888}:${JUPYTER_PORT:-8888}
# Rclone webserver for interactive configuration
- ${RCLONE_PORT_HOST:-53682}:53682
environment:
# Don't enclose values in quotes
- WEBUI_BRANCH=${WEBUI_BRANCH:-master}
- WEBUI_FLAGS=${WEBUI_FLAGS:-}
- WEBUI_PORT=${WEBUI_PORT:-8188}
# Allows running true SSH alongside provider proxy SSH
- SSH_PORT=${SSH_PORT:-22}
- WORKSPACE=${WORKSPACE:-/workspace}
- WORKSPACE_SYNC=${WORKSPACE_SYNC}
- CF_TUNNEL_TOKEN=${CF_TUNNEL_TOKEN:-}
- CF_QUICK_TUNNELS=${CF_QUICK_TUNNELS:-true}
- WEB_ENABLE_AUTH=${WEB_ENABLE_AUTH}
- WEB_USER=${WEB_USER}
- WEB_PASSWORD=${WEB_PASSWORD}
# - PROVISIONING_SCRIPT=${PROVISIONING_SCRIPT:-}