Timothy J. Baek 8 months ago
parent
commit
99764bfd29
1 changed files with 11 additions and 1 deletions
  1. 11 1
      backend/apps/images/main.py

+ 11 - 1
backend/apps/images/main.py

@@ -457,7 +457,17 @@ async def image_generations(
             if form_data.negative_prompt is not None:
             if form_data.negative_prompt is not None:
                 data["negative_prompt"] = form_data.negative_prompt
                 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(
             res = await comfyui_generate_image(
                 app.state.config.MODEL,
                 app.state.config.MODEL,
                 form_data,
                 form_data,