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

Scale inputs before and after motion patch

Browse Source
This commit is contained in:
Eugene Fairley 2025-07-23 17:23:52 -07:00
parent 877ebd9240
commit b3d5b25b69
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes_wan.py
View File

@ -711,12 +711,16 @@ class WanTrackToVideo:
res = res.permute(1,2,3,0)[:, :, :, :3] # T, H, W, C res = res.permute(1,2,3,0)[:, :, :, :3] # T, H, W, C
y = vae.encode(res) y = vae.encode(res)
# Scale latent since patch_motion is non-linear
y = comfy.latent_formats.Wan21().process_in(y)
c = torch.cat((msk, y), dim=1) c = torch.cat((msk, y), dim=1)
res = patch_motion( res = patch_motion(
processed_tracks, c[0], temperature=temperature, topk=topk, vae_divide=(4, 16) processed_tracks, c[0], temperature=temperature, topk=topk, vae_divide=(4, 16)
) )
msk, y = res msk, y = res
y = comfy.latent_formats.Wan21().process_out(y)
msk = -msk + 1.0 # Invert mask to match expected format msk = -msk + 1.0 # Invert mask to match expected format
positive = node_helpers.conditioning_set_values(positive, positive = node_helpers.conditioning_set_values(positive,
{"tracks": processed_tracks, {"tracks": processed_tracks,