mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-09-03 03:17:06 +08:00
save
This commit is contained in:
parent
894ccf4190
commit
af03f5dbe3
@ -672,6 +672,7 @@ class WanTrackToVideo:
|
|||||||
y = vae.encode(
|
y = vae.encode(
|
||||||
res
|
res
|
||||||
)[0]
|
)[0]
|
||||||
|
print("mask shape:", msk.shape, "y shape:", y.shape)
|
||||||
y = torch.concat([msk, y])
|
y = torch.concat([msk, y])
|
||||||
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:]
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user