fixed non controlnet pipe for SDXL

2024-07-25 17:56:31 +02:00
parent 647c691303
commit 77d58c938f
1 changed files with 3 additions and 1 deletions
@@ -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,