From a01a1510027fd2050d665ad7b9f8b8cf7f4abb2d Mon Sep 17 00:00:00 2001
From: LEv145 <40071054+LEv145@users.noreply.github.com>
Date: Mon, 27 Mar 2023 08:59:05 +0200
Subject: [PATCH] Add cpu support and profiles

---
 docker-compose.yml | 56 ++++++++++++++++++++++++++++------------------
 1 file changed, 34 insertions(+), 22 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index 41fb6604b..79351e8b5 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,25 +1,37 @@
 version: "3.9"
 
+
+x-base_service: &base_service
+  image: comfyui:1
+  build: .
+
+  environment:
+    - CLI_ARGS=--listen --port 8188
+
+  ports:
+    - "8188:8188"
+  volumes:
+    - ./models:/opt/comfy_ui/models
+    - ./output:/opt/comfy_ui/output
+    - ./input:/opt/comfy_ui/input
+    - ./custom_nodes:/opt/comfy_ui/custom_nodes
+  
+x-nvidia_docker: &nvidia_docker
+  deploy:
+    resources:
+      reservations:
+        devices:
+          - driver: nvidia
+            device_ids: ["0"]
+            capabilities: [gpu]
+
+
 services:
-  app:
-    image: comfyui:1
-    build: .
-
-    ports:
-      - "8188:8188"
-    volumes:
-      - ./models:/opt/comfy_ui/models
-      - ./output:/opt/comfy_ui/output
-      - ./input:/opt/comfy_ui/input
-      - ./custom_nodes:/opt/comfy_ui/custom_nodes      
-
-    environment:
-      - CLI_ARGS=--listen --port 8188
-
-    deploy:
-      resources:
-        reservations:
-          devices:
-              - driver: nvidia
-                device_ids: ["0"]
-                capabilities: [gpu]
+  nvidia:
+    <<: *base_service
+    <<: *nvidia_docker
+    profiles: ["nvidia"]
+    
+  cpu:
+    <<: *base_service
+    profiles: ["cpu"]