From 1f2662a83b1c9dbdf7a487514ea00c6df369c09d Mon Sep 17 00:00:00 2001 From: Phr00t Date: Sun, 17 Mar 2024 20:14:51 -0400 Subject: [PATCH] batch resizing fix --- nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index 16d39d6..4034dba 100644 --- a/nodes.py +++ b/nodes.py @@ -452,7 +452,7 @@ class DynamiCrafterBatchInterpolation: if out_video.shape[1] != orig_H or out_video.shape[2] != orig_W: out_video = F.interpolate(out_video.permute(0, 3, 1, 2), size=(orig_H, orig_W), mode="bicubic") - out_video = video.permute(0, 2, 3, 1) + out_video = out_video.permute(0, 2, 3, 1) # should we trim middle keyframes? if cut_near_keyframes > 0: