diff --git a/laptop_full_setup.sh b/laptop_full_setup.sh
index d06174f..e418492 100755
--- a/laptop_full_setup.sh
+++ b/laptop_full_setup.sh
@@ -1065,6 +1065,7 @@ $( $SVC_COMFYUI && cat <<'IMGENV'
       - ENABLE_IMAGE_GENERATION=true
       - IMAGE_GENERATION_ENGINE=comfyui
       - COMFYUI_BASE_URL=http://comfyui:8188
+      - COMFYUI_WORKFLOW_NODES=[{"type":"prompt","key":"text","node_ids":["6"]},{"type":"model","key":"ckpt_name","node_ids":["4"]},{"type":"width","key":"width","node_ids":["5"]},{"type":"height","key":"height","node_ids":["5"]},{"type":"steps","key":"steps","node_ids":["3"]},{"type":"seed","key":"seed","node_ids":["3"]}]
 IMGENV
 )
     depends_on:
diff --git a/local-ai-setup.sh b/local-ai-setup.sh
index 5e6fc84..62600d6 100755
--- a/local-ai-setup.sh
+++ b/local-ai-setup.sh
@@ -616,6 +616,7 @@ services:
       - ENABLE_IMAGE_GENERATION=true
       - IMAGE_GENERATION_ENGINE=comfyui
       - COMFYUI_BASE_URL=http://comfyui:8188
+      - COMFYUI_WORKFLOW_NODES=[{"type":"prompt","key":"text","node_ids":["6"]},{"type":"model","key":"ckpt_name","node_ids":["4"]},{"type":"width","key":"width","node_ids":["5"]},{"type":"height","key":"height","node_ids":["5"]},{"type":"steps","key":"steps","node_ids":["3"]},{"type":"seed","key":"seed","node_ids":["3"]}]
     depends_on:
       ollama: {condition: service_healthy}