This commit is contained in:
Eugene Fairley 2025-06-28 23:11:46 -07:00
parent d905e82a9d
commit 085b986e02

View File

@ -1217,9 +1217,9 @@ class WAN21(BaseModel):
res = torch.cat((mask, image), dim=1) res = torch.cat((mask, image), dim=1)
tracks = kwargs.get("tracks", None) tracks = kwargs.get("tracks", None)
if tracks is not None: if tracks is not None:
res = patch_motion(tracks.to(device), res, 220.0, (4, 16), 2)[None] res = patch_motion(tracks.to(device), res[0], 220.0, (4, 16), 2)[None]
return res return res[None]
def extra_conds(self, **kwargs): def extra_conds(self, **kwargs):