batch resizing fix

2024-03-17 20:14:51 -04:00
parent 6ae23034e1
commit 1f2662a83b
1 changed files with 1 additions and 1 deletions
@@ -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: