Add cpu support and profiles

This commit is contained in:
LEv145 2023-03-27 08:59:05 +02:00 committed by GitHub
parent 8ccc455e73
commit a01a151002
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,10 +1,13 @@
version: "3.9" version: "3.9"
services:
app: x-base_service: &base_service
image: comfyui:1 image: comfyui:1
build: . build: .
environment:
- CLI_ARGS=--listen --port 8188
ports: ports:
- "8188:8188" - "8188:8188"
volumes: volumes:
@ -13,9 +16,7 @@ services:
- ./input:/opt/comfy_ui/input - ./input:/opt/comfy_ui/input
- ./custom_nodes:/opt/comfy_ui/custom_nodes - ./custom_nodes:/opt/comfy_ui/custom_nodes
environment: x-nvidia_docker: &nvidia_docker
- CLI_ARGS=--listen --port 8188
deploy: deploy:
resources: resources:
reservations: reservations:
@ -23,3 +24,14 @@ services:
- driver: nvidia - driver: nvidia
device_ids: ["0"] device_ids: ["0"]
capabilities: [gpu] capabilities: [gpu]
services:
nvidia:
<<: *base_service
<<: *nvidia_docker
profiles: ["nvidia"]
cpu:
<<: *base_service
profiles: ["cpu"]