Commit 2b117f9c authored by sandy's avatar sandy Committed by GitHub
Browse files

Fix bug: wan's target_video_len hard code

parent bd1e469c
...@@ -113,7 +113,7 @@ class WanRunner(DefaultRunner): ...@@ -113,7 +113,7 @@ class WanRunner(DefaultRunner):
torch.concat( torch.concat(
[ [
torch.nn.functional.interpolate(img[None].cpu(), size=(h, w), mode="bicubic").transpose(0, 1), torch.nn.functional.interpolate(img[None].cpu(), size=(h, w), mode="bicubic").transpose(0, 1),
torch.zeros(3, 80, h, w), torch.zeros(3, config.target_video_length - 1, h, w),
], ],
dim=1, dim=1,
).cuda() ).cuda()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment