RomanNum3ral
/
ComfyUI
mirror of https://github.com/Comfy-Org/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix 5D reference latent reshaping block

This commit is contained in:
silveroxides 2026-07-13 13:17:41 +02:00
parent 6c163508e1
commit 641ce082a9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/krea2/model.py
View File

@ -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)