This commit is contained in:
Eugene Fairley 2025-06-29 02:02:28 -07:00
parent fdb191c1e6
commit a7794c141e

View File

@ -1218,8 +1218,8 @@ class WAN21(BaseModel):
print(f"image shape: {image.shape}, mask shape: {mask.shape}")
res = torch.cat((mask, image), dim=1)
tracks = kwargs.get("tracks", None)
# if tracks is not None:
# res = patch_motion(tracks.to(device), res[0], 220.0, (4, 16), 2)[None]
if tracks is not None:
res = patch_motion(tracks.to(device), res[0], 220.0, (4, 16), 2)[None]
return res