wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-06-14 11:59:21 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

Fix potential shape missmatch in CFGRenormLongCatImage

Browse Source
This commit is contained in:
Talmaj Marinc 2026-02-23 22:00:59 +01:00
parent 87d0a15307
commit 4fe507248c
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_extras/nodes_longcat_image.py
View File

@ -1,7 +1,11 @@
import logging
import torch import torch
from typing_extensions import override from typing_extensions import override
from comfy_api.latest import ComfyExtension, io from comfy_api.latest import ComfyExtension, io
logger = logging.getLogger(__name__)
class CLIPTextEncodeLongCatImage(io.ComfyNode): class CLIPTextEncodeLongCatImage(io.ComfyNode):
@classmethod @classmethod
@ -61,6 +65,10 @@ class CFGRenormLongCatImage(io.ComfyNode):
B, C, H, W = denoised.shape B, C, H, W = denoised.shape
ps = 2 ps = 2
if H % ps != 0 or W % ps != 0:
logger.warning(f"CFG Renorm: incompatible shape {H}x{W}, skipping renorm")
return denoised
noise = x - denoised noise = x - denoised
noise_cond = x - cond_denoised noise_cond = x - cond_denoised