From 9fecaeb2d5ebbba2bcc370c699d58e63e9be54e8 Mon Sep 17 00:00:00 2001 From: silveroxides Date: Tue, 23 Jun 2026 22:21:39 +0200 Subject: [PATCH] Implement 5D-to-4D temporal latent reshape fix --- comfy/ldm/krea2/model.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/comfy/ldm/krea2/model.py b/comfy/ldm/krea2/model.py index 91a089153..255119cbd 100644 --- a/comfy/ldm/krea2/model.py +++ b/comfy/ldm/krea2/model.py @@ -317,6 +317,9 @@ class SingleStreamDiT(nn.Module): negative_ref_method = ref_latents_method == "negative_index" for ref in ref_latents: + if ref.ndim == 5: + ref_b5, ref_c5, ref_t5, ref_h5, ref_w5 = ref.shape + ref = ref.reshape(ref_b5 * ref_t5, ref_c5, ref_h5, ref_w5) ref_pad = comfy.ldm.common_dit.pad_to_patch_size(ref, (patch, patch)) ref_b, ref_c, ref_h, ref_w = ref_pad.shape ref_gh = ref_h // patch