From 77d58c938fef09d5de7a7be20cf0dbfdb362b308 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ma=C3=ABlys=20Jusseaux?= <https://gitlab.com/mjusseaux>
Date: Thu, 25 Jul 2024 17:56:31 +0200
Subject: [PATCH] fixed non controlnet pipe for SDXL

---
 diffusers_in_comfy.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/diffusers_in_comfy.py b/diffusers_in_comfy.py
index e19edbd..0c73978 100644
--- a/diffusers_in_comfy.py
+++ b/diffusers_in_comfy.py
@@ -142,9 +142,11 @@ class ImageInference:
         return torch.stack([np.transpose(ToTensor()(image), (1, 2, 0)) for image in images])
 
 
-    def infer_image(self, pipeline, seed, steps, cfg, positive, negative, width, height, controlnet_image, controlnet_scale):
+    def infer_image(self, pipeline, seed, steps, cfg, positive, negative, width, height, controlnet_image=None, controlnet_scale=None):
         generator = torch.Generator(device='cuda').manual_seed(seed)
 
+        print(f'at this stage, controlnet image is {controlnet_image} of type {type(controlnet_image)}')
+
         args = {
             "prompt": positive,
             "generator": generator,