fixed image output with 4 channels instead of 3 in diffusers image outpaint

2024-09-28 10:42:08 +02:00
parent a146f10ea8
commit 0b4d11cb15
1 changed files with 2 additions and 2 deletions
@@ -290,7 +290,7 @@ class DiffusersImageOutpaint:
            torch.cuda.ipc_collect()

        last_image = generated_images[-1] # Access the last image
-        image = last_image.convert("RGBA")
+        image = last_image.convert("RGB")
        output=pil2tensor(image)
        
-        return (output,)
+        return (output,)