diff options
Diffstat (limited to 'docker-compose.yml')
| -rw-r--r-- | docker-compose.yml | 30 |
1 files changed, 19 insertions, 11 deletions
diff --git a/docker-compose.yml b/docker-compose.yml index 8c1fa10..1804901 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -2,11 +2,12 @@ services: ollama: image: ollama/ollama:rocm + container_name: ollama restart: always environment: - OLLAMA_MAX_QUEUE: 8192 + OLLAMA_MAX_QUEUE: 512 OLLAMA_MAX_LOADED_MODELS: 1 - OLLAMA_NUM_PARALLEL: 5 + OLLAMA_NUM_PARALLEL: 1 OLLAMA_KEEP_ALIVE: 0 OLLAMA_FLASH_ATTENTION: 1 OLLAMA_GPU_OVERHEAD: 2147483648 @@ -27,31 +28,38 @@ services: - /dev/kfd:/dev/kfd - /dev/dri:/dev/dri - comfy-ui: - image: corundex/comfyui-rocm:latest + comfyui: + image: comfyui + build: + dockerfile: Dockerfile.comfyui + container_name: comfyui restart: always - environment: - CLI_ARGS: --listen 0.0.0.0 + command: > + --listen 0.0.0.0 + --enable-manager + --enable-dynamic-vram + --mmap-torch-files + --multi-user volumes: - type: bind source: ./comfyui/custom_nodes - target: /workspace/ComfyUI/custom_nodes + target: /app/ComfyUI/custom_nodes read_only: false - type: bind source: ./comfyui/input - target: /workspace/ComfyUI/input + target: /app/ComfyUI/input read_only: false - type: bind source: ./comfyui/models - target: /workspace/ComfyUI/models + target: /app/ComfyUI/models read_only: false - type: bind source: ./comfyui/output/ - target: /workspace/ComfyUI/output/ + target: /app/ComfyUI/output/ read_only: false - type: bind source: ./comfyui/user - target: /workspace/ComfyUI/user + target: /app/ComfyUI/user read_only: false ports: - 8188:8188 |