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

save

Browse Source
This commit is contained in:
Eugene Fairley 2025-06-28 00:13:45 -07:00
parent 32553f6d96
commit c254763d6b
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy_extras/nodes_wan.py
View File

@ -659,18 +659,19 @@ class WanTrackToVideo:
zero_frames = torch.zeros(3, 81 - 1, height, width) zero_frames = torch.zeros(3, 81 - 1, height, width)
start_image = start_image.permute(3,0,1,2) * 2 - 1 # C, T, H, W start_image = start_image.permute(3,0,1,2) * 2 - 1 # C, T, H, W
res = torch.concat([ res = torch.concat([
start_image.to(start_image.device), start_image.to(start_image.device),
zero_frames zero_frames
], ],
dim=1).to(start_image.device) dim=1).to(start_image.device)
print("start vid:", res.permute(1,2,3,0)[:, :, :, :3])
y = vae.encode( y = vae.encode(
res.permute(1,2,3,0)[:, :, :, :3] # T, H, W, C res.permute(1,2,3,0)[:, :, :, :3] # T, H, W, C
)[0] )[0]
y = torch.concat([msk, y]) y = torch.concat([msk, y])
print('?')
motion_patched = patch_motion(processed_tracks, y, temperature, (4, 16), topk)[None] motion_patched = patch_motion(processed_tracks, y, temperature, (4, 16), topk)[None]
mask, video = motion_patched[:, 0:4], motion_patched[:, 4:] mask, video = motion_patched[:, 0:4], motion_patched[:, 4:]
# Add motion features to conditioning # Add motion features to conditioning