-
Notifications
You must be signed in to change notification settings - Fork 84
/
comfy-docker-compose.yml
48 lines (46 loc) · 1.81 KB
/
comfy-docker-compose.yml
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
version: "3.8"
services:
selenium:
container_name: ${SELENIUM_CONTAINER_NAME}
image: ${ACR_ORG}/${SELENIUM_IMAGE}
network_mode: host
shm_size: 2g
command: sleep 5400
entrypoint: /opt/bin/entry_point.sh
restart: "no"
onediff-test:
container_name: ${CONTAINER_NAME}
image: ${ACR_ORG}/${MATRIX_IMAGE}
command: sleep 5400
privileged: true
shm_size: 8g
network_mode: host
pids_limit: 2000
cap_add:
- SYS_PTRACE
security_opt:
- seccomp=unconfined
environment:
HF_HUB_OFFLINE: "1"
ONEFLOW_MLIR_ENABLE_TIMING: "1"
ONEFLOW_MLIR_PRINT_STATS: "1"
CI: "1"
SILICON_ONEDIFF_LICENSE_KEY: ${SILICON_ONEDIFF_LICENSE_KEY}
volumes:
- $HOME/test-container-cache-${CONTAINER_NAME}/dot-local:/root/.local
- $HOME/test-container-cache-${CONTAINER_NAME}/dot-cache:/root/.cache
- /share_nfs:/share_nfs:ro
- ${PWD}/${COMFYUI_SRC_DIR}:/app/ComfyUI
- /share_nfs/hf_models/comfyui_resources/custom_nodes/ComfyUI_IPAdapter_plus:/app/ComfyUI/custom_nodes/ComfyUI_IPAdapter_plus
- /share_nfs/hf_models/comfyui_resources/input/input_image_vermeer.png:/app/ComfyUI/input/input_image_vermeer.png:ro
- /share_nfs/hf_models/comfyui_resources/input/a_car.png:/app/ComfyUI/input/a_car.png:ro
- ${PWD}/onediff_comfy_nodes:/app/ComfyUI/custom_nodes/onediff_comfy_nodes
- ${SDXL_BASE}:/app/ComfyUI/models/checkpoints/sd_xl_base_1.0.safetensors:ro
- ${UNET_INT8}:/app/ComfyUI/models/unet_int8/unet_int8:ro
- ${CONTROL_LORA_OPENPOSEXL2_RANK256}:/app/ComfyUI/models/controlnet/control-lora-openposeXL2-rank256.safetensors:ro
- $PWD/tests/comfyui/workflows/input/control:/app/ComfyUI/input/control:ro
- $PWD:/src/onediff
working_dir: /src/onediff
restart: "no"
depends_on:
- selenium