xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-03 05:37:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

'save'

Browse Source
This commit is contained in:
Eugene Fairley 2025-06-28 23:07:56 -07:00
parent e643977d61
commit d905e82a9d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes_wan.py
View File

@ -655,7 +655,7 @@ class WanTrackToVideo:
msk = msk.view(1, msk.shape[1] // 4, 4, lat_h, lat_w) msk = msk.view(1, msk.shape[1] // 4, 4, lat_h, lat_w)
# first batch # first batch
msk = msk.transpose(1, 2)[0] msk = msk.transpose(1, 2)
zero_frames = torch.zeros(3, 81 - 1, height, width) zero_frames = torch.zeros(3, 81 - 1, height, width)
@ -672,7 +672,7 @@ class WanTrackToVideo:
y = vae.encode( y = vae.encode(
res res
)[0] )
print("mask shape:", msk.shape, "y shape:", y.shape) print("mask shape:", msk.shape, "y shape:", y.shape)
# y = torch.concat([msk, y]) # y = torch.concat([msk, y])