From cf00f9b704b2d516f1fd84707bf0031b0e85c300 Mon Sep 17 00:00:00 2001 From: ComplexRobot Date: Sun, 16 Mar 2025 07:05:02 -0700 Subject: [PATCH] Improve minor logical error in the blur/sharpen algorithm. --- nodes.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nodes.py b/nodes.py index cd0a8dd..0044e62 100644 --- a/nodes.py +++ b/nodes.py @@ -35,12 +35,12 @@ class Simple_Frame_Interpolation: if gaussian_blur and blur_radius != 0 and blur_sigma != 0: B2, H2, W2, C2 = images.shape - images.flatten(1, 2).unsqueeze(2) + images.flatten(0, 2).unflatten(0, (W2, H2, 1)) if blur_sigma > 0: images, = Blur.blur(self, images, blur_radius, blur_sigma) elif blur_sigma < 0 : images, = Sharpen.sharpen(self, images, blur_radius, -blur_sigma, sharpen_alpha) - images.flatten(1, 2).unflatten(1, (H2, W2)) + images.flatten(0, 2).unflatten(0, (1, H2, W2)) images = images.flatten(0, 2).unflatten(0, (new_frame_count, H, W))