From 99764bfd29b38f26684f1f096ba6a8675f4bf098 Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Wed, 21 Aug 2024 14:49:17 +0200
Subject: [PATCH] fix

---
 backend/apps/images/main.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/backend/apps/images/main.py b/backend/apps/images/main.py
index 8bd88308b..ef5f1418f 100644
--- a/backend/apps/images/main.py
+++ b/backend/apps/images/main.py
@@ -457,7 +457,17 @@ async def image_generations(
             if form_data.negative_prompt is not None:
                 data["negative_prompt"] = form_data.negative_prompt
 
-            form_data = ComfyUIGenerateImageForm(**data)
+            form_data = ComfyUIGenerateImageForm(
+                {
+                    "workflow": ComfyUIWorkflow(
+                        {
+                            "workflow": app.state.config.COMFYUI_WORKFLOW,
+                            "nodes": app.state.config.COMFYUI_WORKFLOW_NODES,
+                        }
+                    ),
+                    **data,
+                }
+            )
             res = await comfyui_generate_image(
                 app.state.config.MODEL,
                 form_data,