From a8f2cad1b6a54fdb67e0691d5af8ec47c4a5444e Mon Sep 17 00:00:00 2001
From: Derek Holloway <derek@mistox.com>
Date: Fri, 4 Jul 2025 19:38:06 +0000
Subject: [PATCH] Add Volumes

---
 docker-compose.yml | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index 1012af8..3e7eca9 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -2,6 +2,9 @@ services:
   comfyui:
     build: .
     container_name: ai-comfyui
+    volumes:
+      - ./data/models:/app/comfyui/models
+      - ./data/output:/app/comfyui/output
     ports:
       - 8188:8188
     restart: always
@@ -11,5 +14,4 @@ services:
           devices:
             - driver: nvidia
               count: all
-              capabilities: [gpu]
-
+              capabilities: [gpu]
\ No newline at end of file