From 0addfc6101f7a834c7fb6e0a1b26529360ab5350 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Sat, 29 Mar 2025 12:51:49 +0200 Subject: [PATCH] Maybe fix shape mismatch with SLG and I2V when doing batched cfg --- nodes/model_optimization_nodes.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/nodes/model_optimization_nodes.py b/nodes/model_optimization_nodes.py index 9072bcd..cafc8a0 100644 --- a/nodes/model_optimization_nodes.py +++ b/nodes/model_optimization_nodes.py @@ -1123,10 +1123,10 @@ class SkipLayerGuidanceWanVideo: prev_img_uncond = args["img"][0].unsqueeze(0) new_args = { - "img": args["img"][1], - "txt": args["txt"][1], - "vec": args["vec"][1], - "pe": args["pe"][1] + "img": args["img"][1].unsqueeze(0), + "txt": args["txt"][1].unsqueeze(0), + "vec": args["vec"][1].unsqueeze(0), + "pe": args["pe"][1].unsqueeze(0) } block_out = original_block(new_args)