xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-09 21:04:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Maybe fix shape mismatch with SLG and I2V when doing batched cfg

Browse Source
This commit is contained in:
kijai 2025-03-29 12:51:49 +02:00
parent 916461c432
commit 0addfc6101
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
nodes/model_optimization_nodes.py
View File

@ -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)