Merge pull request #172 from Duanyll/master

fix: PainterNode buffer is not large enough
2025-08-13 07:57:13 +03:00
parent 8aa9c0b30d c781e3f858
commit 319ae79c85
1 changed files with 3 additions and 0 deletions
@@ -382,6 +382,9 @@ class PainterNode(object):
            input_images = []

            for imgs in images:
+                # Add alpha channel if not present
+                if imgs.shape[2] == 3:
+                    imgs = torch.cat([imgs, torch.ones((*imgs.shape[:2], 1))], dim=2)
                i = 255.0 * imgs.cpu().numpy()
                i = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8), mode="RGBA")
                input_images.append(toBase64ImgUrl(i))