diff --git a/layered_diffusion.py b/layered_diffusion.py index 5ffc79b..6e2e90d 100644 --- a/layered_diffusion.py +++ b/layered_diffusion.py @@ -129,7 +129,10 @@ class LayeredDiffusionDecodeRGBA(LayeredDiffusionDecode): def decode(self, samples, images, sd_version: str, sub_batch_size: int): image, mask = super().decode(samples, images, sd_version, sub_batch_size) alpha = 1.0 - mask - return JoinImageWithAlpha().join_image_with_alpha(image, alpha) + if hasattr(JoinImageWithAlpha, 'execute') and callable(JoinImageWithAlpha.execute): + return JoinImageWithAlpha().execute(image, alpha) + else: + return JoinImageWithAlpha().join_image_with_alpha(image, alpha) class LayeredDiffusionDecodeSplit(LayeredDiffusionDecodeRGBA):