fix issues with the ruff style check

This commit is contained in:
lzyhha 2025-02-16 11:43:04 +00:00
parent 7ed909f457
commit 1c4bbc2c53

View File

@ -25,7 +25,7 @@ class RenormCFG:
if timestep[0] < cfg_trunc:
cond_eps, uncond_eps = cond_denoised[:, :in_channels], uncond_denoised[:, :in_channels]
cond_rest, uncond_rest = cond_denoised[:, in_channels:], uncond_denoised[:, in_channels:]
cond_rest, _ = cond_denoised[:, in_channels:], uncond_denoised[:, in_channels:]
half_eps = uncond_eps + cond_scale * (cond_eps - uncond_eps)
half_rest = cond_rest
@ -41,7 +41,7 @@ class RenormCFG:
half_eps = half_eps * (max_new_norm / new_pos_norm)
else:
cond_eps, uncond_eps = cond_denoised[:, :in_channels], uncond_denoised[:, :in_channels]
cond_rest, uncond_rest = cond_denoised[:, in_channels:], uncond_denoised[:, in_channels:]
cond_rest, _ = cond_denoised[:, in_channels:], uncond_denoised[:, in_channels:]
half_eps = cond_eps
half_rest = cond_rest
@ -94,7 +94,7 @@ class CLIPTextEncodeLumina2(ComfyNodeABC):
NODE_CLASS_MAPPINGS = {
"CLIPTextEncodeLumina": CLIPTextEncodeLumina2,
"CLIPTextEncodeLumina2": CLIPTextEncodeLumina2,
"RenormCFG": RenormCFG
}