wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-19 04:48:17 +08:00
Code Issues Actions 35 Packages Projects Releases Wiki Activity

Implement 5D-to-4D temporal latent reshape fix

Browse Source
This commit is contained in:
silveroxides 2026-06-23 22:21:39 +02:00
parent 64ed140c89
commit 9fecaeb2d5
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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