From 2968761f43a2dc638269e31cb71493eeb022b031 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Mon, 18 Mar 2024 02:36:04 +0200 Subject: [PATCH] Update nodes.py --- nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index f7a4278..0e0b577 100644 --- a/nodes.py +++ b/nodes.py @@ -451,7 +451,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) last_image = out_video[-1].unsqueeze(0) return (out_video, last_image)