From 641ce082a93d1a502dede8f61984318cca6045ef Mon Sep 17 00:00:00 2001 From: silveroxides Date: Mon, 13 Jul 2026 13:17:41 +0200 Subject: [PATCH] Fix 5D reference latent reshaping block --- comfy/ldm/krea2/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/ldm/krea2/model.py b/comfy/ldm/krea2/model.py index f9bc74b1c..d4081eb7c 100644 --- a/comfy/ldm/krea2/model.py +++ b/comfy/ldm/krea2/model.py @@ -368,7 +368,7 @@ class SingleStreamDiT(nn.Module): for i, ref in enumerate(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 = ref.movedim(2, 1).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_pad = comfy.utils.repeat_to_batch_size(ref_pad, bs) ref_pad = ref_pad.to(device=device, dtype=dtype)