From 7a5aff1946e3e217166905ccf17a6bf41f14eaa4 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Thu, 14 May 2026 12:52:16 +0300 Subject: [PATCH] Slice alpha channel --- comfy_extras/nodes_moge.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/comfy_extras/nodes_moge.py b/comfy_extras/nodes_moge.py index 436ca27ea..d8bcedf10 100644 --- a/comfy_extras/nodes_moge.py +++ b/comfy_extras/nodes_moge.py @@ -141,6 +141,7 @@ class MoGePanoramaInference(io.ComfyNode): if image.shape[0] != 1: raise ValueError(f"MoGePanoramaInference takes a single image (got batch of {image.shape[0]})") + image = image[..., :3] H, W = int(image.shape[1]), int(image.shape[2]) scale = min(merge_resolution / max(H, W), 1.0) merge_h, merge_w = max(int(H * scale), 32), max(int(W * scale), 32) @@ -257,6 +258,7 @@ class MoGeInference(io.ComfyNode): @classmethod def execute(cls, moge_model, image, resolution_level, fov_x_degrees, batch_size, force_projection, apply_mask) -> io.NodeOutput: + image = image[..., :3] bchw = image.movedim(-1, -3).contiguous() B = bchw.shape[0] fov = None if fov_x_degrees <= 0 else float(fov_x_degrees)