feat: simplified and added compatibility with Windows

blib-la · Nov 19, 2024 · 9f41231 · 9f41231
1 parent 697d985
commit 9f41231
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -1,16 +1,18 @@
-version: "3.8"
-
 services:
-  comfyui:
-    image: timpietruskyblibla/runpod-worker-comfy:dev
-    container_name: comfyui-worker
+  comfyui-worker:
+    image: timpietruskyblibla/runpod-worker-comfy:latest-flux1-schnell
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
     environment:
-      - NVIDIA_VISIBLE_DEVICES=all
       - SERVE_API_LOCALLY=true
     ports:
       - "8000:8000"
       - "8188:8188"
-    runtime: nvidia
     volumes:
       - ./data/comfyui/output:/comfyui/output
       - ./data/runpod-volume:/runpod-volume