From 5a36a4c866e6f3552f646aea05ddc3bd5433a125 Mon Sep 17 00:00:00 2001
From: Derek Holloway <derek@mistox.com>
Date: Mon, 7 Jul 2025 03:52:03 +0000
Subject: [PATCH] Add build args

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

diff --git a/docker-compose.yml b/docker-compose.yml
index 3e7eca9..76484c3 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,10 +1,18 @@
 services:
-  comfyui:
-    build: .
+   comfyui:
     container_name: ai-comfyui
+    build:
+      context: .
+      dockerfile: dockerfile-comfyui
+      args:
+        ComfyUI_Release: "0.3.43" # Must be (Maj.Min.Patch) format. -> https://github.com/comfyanonymous/ComfyUI/releases
+        Python_Release: "3.12.11" # Must be (Maj.Min.Patch) format. -> https://www.python.org/downloads/source/
+        Pytorch_Release: "2.7.1" # Must be (Maj.Min.Patch) format. -> # https://github.com/pytorch/pytorch/releases
+        Pytorch_Jobs: 12 # Set the number of jobs for the Pytorch_Build -> (Gb of RAM)/17 rounded down to the nearest whole number
+        CUDA_ARCH_LIST: "8.9" # Set the CUDA arch to compile for -> get the current cuda version from 'nvidia-smi'
     volumes:
-      - ./data/models:/app/comfyui/models
-      - ./data/output:/app/comfyui/output
+      - ./data/comfyui/models:/app/comfyui/models
+      - ./data/comfyui/output:/app/comfyui/output
     ports:
       - 8188:8188
     restart: always